logstash-input-rethinkdb 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: ee3f3933a77ac91a562be9e4a09a70663a456382
+  data.tar.gz: a8170147fd7d527903a6bb6cc46ca40518973bbb
+SHA512:
+  metadata.gz: da1a6acba6a759f9533c3ae53473a80f1be1a2a4e0370ca0e69c3f420d6c06caf31c5b1b51f4ef2126e99b7b65c20b317a5ceb5208b6fdb00d4814595f58ddfd
+  data.tar.gz: fe318a16b281d0558bf08aa3e504ed40ef437ab9c4282d050c230397465ca129dbfce5977f99905292de3334fab2ae333c204f3dfce073b0044b035284685d0d

data/.gitignore

@@ -0,0 +1,5 @@
+*.gem
+Gemfile.lock
+Gemfile.bak
+.bundle
+vendor

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+gem "eventmachine"
+gem "rethinkdb", :path => "/home/josh/Code/rethinkdb/drivers/ruby"
+gemspec

data/LICENSE

@@ -0,0 +1,13 @@
+Copyright (c) 2012-2015 Elasticsearch <http://www.elasticsearch.org>
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

data/README.md

@@ -0,0 +1,86 @@
+# RethinkDB Logstash input plugin
+
+This is a plugin for [Logstash](https://github.com/elasticsearch/logstash).
+
+It is fully free and fully open source. The license is Apache 2.0, meaning you are pretty much free to use it however you want in whatever way.
+
+## Documentation
+
+Logstash provides infrastructure to automatically generate documentation for this plugin. We use the asciidoc format to write documentation so any comments in the source code will be first converted into asciidoc and then into html. All plugin documentation are placed under one [central location](http://www.elasticsearch.org/guide/en/logstash/current/).
+
+- For formatting code or config example, you can use the asciidoc `[source,ruby]` directive
+- For more asciidoc formatting tips, see the excellent reference here https://github.com/elasticsearch/docs#asciidoc-guide
+
+## Need Help?
+
+Need help? Try #logstash on freenode IRC or the logstash-users@googlegroups.com mailing list.
+
+## Developing
+
+### 1. Plugin Developement and Testing
+
+#### Code
+- To get started, you'll need JRuby with the Bundler gem installed.
+
+- Create a new plugin or clone and existing from the GitHub [logstash-plugins](https://github.com/logstash-plugins) organization. We also provide [example plugins](https://github.com/logstash-plugins?query=example).
+
+- Install dependencies
+```sh
+bundle install
+```
+
+#### Test
+
+- Update your dependencies
+
+```sh
+bundle install
+```
+
+- Run tests
+
+```sh
+bundle exec rspec
+```
+
+### 2. Running your unpublished Plugin in Logstash
+
+#### 2.1 Run in a local Logstash clone
+
+- Edit Logstash `Gemfile` and add the local plugin path, for example:
+```ruby
+gem "logstash-filter-awesome", :path => "/your/local/logstash-filter-awesome"
+```
+- Install plugin
+```sh
+bin/plugin install --no-verify
+```
+- Run Logstash with your plugin
+```sh
+bin/logstash -e 'filter {awesome {}}'
+```
+At this point any modifications to the plugin code will be applied to this local Logstash setup. After modifying the plugin, simply rerun Logstash.
+
+#### 2.2 Run in an installed Logstash
+
+You can use the same **2.1** method to run your plugin in an installed Logstash by editing its `Gemfile` and pointing the `:path` to your local plugin development directory or you can build the gem and install it using:
+
+- Build your plugin gem
+```sh
+gem build logstash-filter-awesome.gemspec
+```
+- Install the plugin from the Logstash home
+```sh
+bin/plugin install /your/local/plugin/logstash-filter-awesome.gem
+```
+- Start Logstash and proceed to test the plugin
+
+## Contributing
+
+All contributions are welcome: ideas, patches, documentation, bug reports, complaints, and even something you drew up on a napkin.
+
+Programming is not a required skill. Whatever you've seen about open source and maintainers or community members  saying "send patches or die" - you will not see that here.
+
+It is more important to the community that you are able to contribute.
+
+For more information about contributing, see the [CONTRIBUTING](https://github.com/elasticsearch/logstash/blob/master/CONTRIBUTING.md) file.

data/Rakefile

@@ -0,0 +1 @@
+require "logstash/devutils/rake"

data/lib/logstash/inputs/rethinkdb.rb

@@ -0,0 +1,230 @@
+# encoding: utf-8
+
+require "logstash/inputs/base"
+require "logstash/namespace"
+require "eventmachine"
+require "rethinkdb"
+
+class LogStash::Inputs::RethinkDB < LogStash::Inputs::Base
+  config_name "rethinkdb"
+  default :codec, "json_lines"
+  attr_accessor :logger
+
+  include RethinkDB::Shortcuts
+
+  # Hostname of RethinkDB server
+  config :host, :validate => :string, :default => "localhost"
+  # Driver connection port of RethinkDB server
+  config :port, :validate => :number, :default => 28015
+  # Auth key of RethinkDB server (don't provide if nil)
+  config :auth_key, :validate => :string, :default => ""
+  # Time period to squash changefeeds on. Defaults to no squashing.
+  config :squash, :default => false
+  # Which tables to watch for changes
+  config :watch_tables, :validate => :array, :default => []
+  # Which databases to watch for changes. Tables added or removed from
+  # these databases will be watched or unwatched accordingly
+  config :watch_dbs, :validate => :array, :default => []
+
+
+  # Part of the logstash input interface
+  def register
+    # {db => {table => QueryHandle}}
+    @table_feeds = Hash.new { |hsh, key| hsh[key] = {} }
+    # {db => QueryHandle}
+    @db_feeds = {}
+    @queue = nil
+  end
+
+  # # Part of the logstash input interface
+  def run(queue)
+    @queue = queue
+    @conn = r.connect(
+      :host => @host,
+      :port => @port,
+      :auth_key => @auth_key,
+    )
+    EM.run do
+      @logger.log "Eventmachine loop started"
+      @watch_dbs.uniq.each do |db|
+        create_db_feed(db, DBHandler.new(db, self))
+      end
+      @watch_tables.uniq.each do |db_table|
+        db, table = db_table.split '.'
+        db, table = "test", db if table.nil?
+        update_db_tables(nil, {'db' => db, 'name' => table})
+      end
+    end
+  end
+
+  def send(db, table, old_val, new_val)
+    event = LogStash::Event.new(
+      "db" => db,
+      "table" => table,
+      "old_val" => old_val,
+      "new_val" => new_val
+    )
+    decorate(event)
+    @queue << event
+  end
+
+  def update_db_tables(old_val, new_val)
+    unless new_val.nil?
+      handler = TableHandler.new(new_val['db'], new_val['name'], self)
+      create_table_feed(new_val['db'], new_val['name'], handler)
+    end
+    unless old_val.nil?
+      unregister_table(old_val['db'], old_val['name'], nil)
+    end
+  end
+
+  def register_table(db, table, qhandle)
+    # Add a table feed to the registry
+    unless @table_feeds.has_key?(db) &&
+           @table_feeds[db].has_key?(table)
+      @logger.log("Watching table #{db}.#{table}")
+      @table_feeds[db][table] = qhandle
+    else
+      qhandle.close
+    end
+  end
+
+  def unregister_table(db, table, qhandle)
+    # Remove a table from the registry.
+    if @table_feeds.has_key?(db) &&
+       @table_feeds[db].has_key?(table) &&
+       # If a duplicate feed comes in for the same table and needs to
+       # be unregistered, we need to check if the handle is the same
+       (qhandle.nil? || @table_feeds[db][table].equal?(qhandle))
+      @logger.log("Unregistering table #{db}.#{table}")
+      @table_feeds[db].delete(table).close
+    end
+  end
+
+  def register_db(db, qhandle)
+    # Add a db to the registry to watch it for updates to which tables
+    # are listed in it.
+    unless @db_feeds.has_key? db
+      @db_feeds[db] = qhandle
+      @logger.log "Feed for db '#{db}' registered"
+    end
+  end
+
+  def unregister_db(db)
+    # Remove a db from the registry, close all of its feeds, and
+    # remove its entry in @db_admin_tables
+    if @table_feeds.has_key?(db)
+      @logger.log("Unregistering feed for db '#{db}'")
+      @table_feeds[db].keys.each do |table|
+        unregister_table(db, table, nil)
+      end
+      @table_feeds.delete(db)
+      @db_feeds.delete(db).close
+    end
+  end
+
+  def create_db_feed(db, handler)
+    r.db('rethinkdb').
+      table('table_status').
+      filter {|row| row['status']['all_replicas_ready']}.
+      pluck('db', 'name').
+      changes(:squash => @squash, :include_states => true).
+      em_run(@conn, handler)
+  end
+
+  def create_table_feed(db, table, handler)
+    r.db(db).
+      table(table).
+      changes(:squash => @squash, :include_states => true).
+      em_run(@conn, handler)
+  end
+
+  def teardown
+    # Goes through all existing handles and closes them, then clears
+    # out the registry and closes the connection to RethinkDB
+    @table_feeds.values.each do |tables|
+      tables.values.each { |qhandle| qhandle.close }
+    end
+    @db_feeds.values.each { |qhandle| qhandle.close }
+    @table_feeds.clear
+    @db_feeds.clear
+    @conn.close
+    EM.stop
+    @queue = nil
+  end
+
+end
+
+# This handles feeds listening for changes to documents in a table
+class TableHandler < RethinkDB::Handler
+  attr_accessor :db
+  attr_accessor :table
+  def initialize(db, table, plugin)
+    super()
+    @db = db
+    @table = table
+    @plugin = plugin
+  end
+
+  def on_initial(val)
+    @plugin.send(@db, @table, nil, val)
+  end
+
+  def on_change(old_val, new_val)
+    @plugin.send(@db, @table, old_val, new_val)
+  end
+
+  def on_open(qhandle)
+    @plugin.register_table(@db, @table, qhandle)
+  end
+
+  def on_close(qhandle)
+    @plugin.unregister_table(@db, @table, qhandle)
+  end
+
+  def on_error(err, qhandle)
+    @plugin.logger.error(err.to_s)
+    @plugin.unregister_table(@db, @table, qhandle)
+  end
+
+  def on_change_error(err_str)
+    @plugin.logger.warn(err.to_s)
+  end
+end
+
+# Handler for changes to the tables in a database
+class DBHandler < RethinkDB::Handler
+
+  attr_accessor :db
+
+  def initialize(db, plugin)
+    super()
+    @db = db
+    @plugin = plugin
+  end
+
+  def on_open(qhandle)
+    @plugin.register_db(@db, qhandle)
+  end
+
+  def on_close(qhandle)
+    @plugin.unregister_db(@db)
+  end
+
+  def on_error(err, qhandle)
+    @plugin.logger.error(err.to_s)
+    @plugin.unregister_db(@db)
+  end
+
+  def on_change_error(err_str)
+    @plugin.logger.warn(err_str)
+  end
+
+  def on_initial_val(val)
+    @plugin.update_db_tables(nil, val)
+  end
+
+  def on_change(old_val, new_val)
+    @plugin.update_db_tables(old_val, new_val)
+  end
+end

data/logstash-input-rethinkdb.gemspec

@@ -0,0 +1,26 @@
+Gem::Specification.new do |s|
+  s.name = 'logstash-input-rethinkdb'
+  s.version         = '0.1.0'
+  s.licenses = ['Apache License (2.0)']
+  s.summary = "Streams changes from RethinkDB tables into LogStash."
+  s.description = "This gem is a logstash plugin required to be installed on top of the Logstash core pipeline using $LS_HOME/bin/plugin install gemname. This gem is not a stand-alone program"
+  s.authors = ["RethinkDB"]
+  s.email = 'josh@rethinkdb.com'
+  s.homepage = "http://www.rethinkdb.com/docs/logstash-plugins"
+  s.require_paths = ["lib"]
+
+  # Files
+  s.files = `git ls-files`.split($\)
+  # Tests
+  s.test_files = s.files.grep(%r{^(test|spec|features)/})
+
+  # Special flag to let us know this is actually a logstash plugin
+  s.metadata = { "logstash_plugin" => "true", "logstash_group" => "input" }
+
+  # Gem dependencies
+  s.add_runtime_dependency "logstash-core", '>= 1.4.0', '< 2.0.0'
+  s.add_runtime_dependency 'logstash-codec-plain'
+  s.add_runtime_dependency 'rethinkdb', '~>2.0.0'
+  s.add_runtime_dependency 'eventmachine', '~>1.0.7'
+  s.add_development_dependency 'logstash-devutils'
+end

data/spec/inputs/test_input.rb

@@ -0,0 +1 @@
+require "logstash/devutils/rspec/spec_helper"

metadata

@@ -0,0 +1,130 @@
+--- !ruby/object:Gem::Specification
+name: logstash-input-rethinkdb
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- RethinkDB
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2015-04-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.4.0
+    - - <
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  name: logstash-core
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.4.0
+    - - <
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: logstash-codec-plain
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  name: rethinkdb
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.7
+  name: eventmachine
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.7
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: logstash-devutils
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: This gem is a logstash plugin required to be installed on top of the Logstash core pipeline using $LS_HOME/bin/plugin install gemname. This gem is not a stand-alone program
+email: josh@rethinkdb.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/logstash/inputs/rethinkdb.rb
+- logstash-input-rethinkdb.gemspec
+- spec/inputs/test_input.rb
+homepage: http://www.rethinkdb.com/docs/logstash-plugins
+licenses:
+- Apache License (2.0)
+metadata:
+  logstash_plugin: 'true'
+  logstash_group: input
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project:
+rubygems_version: 2.1.9
+signing_key:
+specification_version: 4
+summary: Streams changes from RethinkDB tables into LogStash.
+test_files:
+- spec/inputs/test_input.rb