fluent-plugin-cratedb 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 827ea567aeac08e5e0f453e1fa72b5fd0b168f69
+  data.tar.gz: 2dc49370a61a21f3d22a353b0112472928ffd0dc
+SHA512:
+  metadata.gz: 85ee563ce046878641a8687b231d5b926d63be76da312ee78e1b1c7284799759f84f5dc0cf055830bdeb9cb728d30863efac6430e43ef264f1fbe8955afc406a
+  data.tar.gz: 252fe5d3c55e43714472d448c4265dead1dbac9c510316ba0152c241fde205d06aea13648a70835dddb022759937b74600b0f45aac5d2ed04832236905929bfb

data/.gitignore

@@ -0,0 +1,11 @@
+.bundle/
+.yardoc
+Gemfile.lock
+_yardoc/
+coverage/
+doc/
+pkg/
+spec/reports/
+tmp/
+bin/
+*.gem

data/.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+  - 2.1.10
+before_install: gem install bundler -v 1.10.6

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in fluent-plugin-cratedb.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,13 @@
+Copyright (c) 2016 - BUOM
+
+   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,57 @@
+# Fluent::Plugin::CrateDB, a plugin for [Fluentd](http://fluentd.org)
+
+fluent-plugin-cratedb is a buffered output plugin for fluentd and CrateDB.
+
+
+## Installation
+
+    $ fluent-gem install fluent-plugin-cratedb
+
+## Usage
+
+`type` `createdb`
+
+--------------
+
+**Options:**
+
+`host`: cratedb host (default: 127.0.0.1)
+
+`port`: cratedb port (default: 4200)
+
+`hosts`: an array of servers including ports [127.0.0.1:4200, 10.0.0.1:4201]
+
+`table`: bulk insert table (require)
+
+`column_names`: bulk insert column (require)
+
+`key_names`: key name values (default `column_names`)
+
+
+## Configuration Example
+
+```
+<match mylog.*>
+  @type cratedb
+  host  localhost
+  port  4200
+  column_names id,user_name,created_at,updated_at
+  table users
+  flush_interval 10s
+</match>
+```
+
+## Contributing
+
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request
+
+
+## Licence
+
+
+This package was distributed under Apache-2.0 licence, see LICENCE file for details.

data/Rakefile

@@ -0,0 +1,10 @@
+require "bundler/gem_tasks"
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << "test"
+  t.libs << "lib"
+  t.test_files = FileList['test/**/*_test.rb']
+end
+
+task :default => :test

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/fluent-plugin-cratedb.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+#require 'fluent/plugin/cratedb/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "fluent-plugin-cratedb"
+  spec.version       = File.read('VERSION').strip
+  spec.authors       = ["buom"]
+  spec.email         = ["me@buom.io"]
+
+  spec.summary       = %q{A plugin for Fluentd}
+  spec.description   = %q{Fluent Output Plugin for CrateDB (http://crate.io)}
+  spec.homepage      = "https://github.com/buom/fluent-plugin-cratedb"
+  spec.license       = "Apache-2.0"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features|bin)/}) }
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "crate_ruby", "~> 0.0.8"
+  spec.add_runtime_dependency "fluentd", "~> 0.12.0"
+
+  spec.add_development_dependency "rake", "~> 10.0"
+  #spec.add_development_dependency "minitest"
+end

data/lib/fluent/plugin/out_cratedb.rb

@@ -0,0 +1,114 @@
+require 'crate_ruby'
+require 'digest'
+require 'fluent/output'
+
+
+module Fluent
+  class Fluent::CratedbOutput < Fluent::BufferedOutput
+    Fluent::Plugin.register_output('cratedb', self)
+
+    include Fluent::SetTimeKeyMixin
+
+    config_param :host, :string, :default => 'localhost', :desc => "CrateDB host."
+    config_param :port, :integer, :default => 4200, :desc => "CrateDB port."
+    config_param :hosts, :array, :default => nil
+    config_param :http_options, :hash, :default => {}
+    config_param :column_names, :string, :desc => "Bulk insert column."
+    config_param :key_names, :string, :default => nil
+    config_param :table, :string, :desc => "Bulk insert table."
+
+    # Define `log` method for v0.10.42 or earlier
+    unless method_defined?(:log)
+      define_method("log") { $log }
+    end
+
+    def configure(conf)
+      super
+
+      if @column_names.nil?
+        fail Fluent::ConfigError, 'column_names MUST specified, but missing'
+      end
+
+      @column_names = @column_names.split(',').collect(&:strip)
+      @key_names = @key_names.nil? ? @column_names : @key_names.split(',').collect(&:strip)
+ 
+      @servers = @hosts ? @hosts : ["#{@host}:#{@port}"]
+    end
+
+    def start
+      super
+      require 'logger'
+      opts = {:http_options => @http_options, :logger => Logger.new(STDERR)}
+      @client = CrateClient.new(@servers, opts)
+    end
+
+    def shutdown
+      super
+    end
+
+    def format(tag, time, record)
+      [tag, time, format_proc.call(tag, time, record)].to_msgpack
+    end
+
+    def write(chunk)
+      values = []
+      chunk.msgpack_each do |tag, time, data|
+        #data = format_proc.call(tag, time, data)
+        values << data
+      end
+      sql = "INSERT INTO #{@table} (#{@column_names.join(',')}) VALUES (#{ @column_names.map { |key| '?' }.join(',') })"
+      #log.info(sql)
+      @client.execute(sql, nil, values, @http_options)
+
+      #digest = Digest::SHA1.hexdigest(data)
+      #@client.blob_put(table_name, digest, data)
+    end
+
+
+    private
+
+    def format_proc
+      proc do |tag, time, record|
+        values = []
+        @key_names.each_with_index do |key, i|
+          value = record[key]
+          values << value
+        end
+        values
+      end
+    end
+  end
+
+  class CrateClient < CrateRuby::Client
+
+    def initialize(servers = [], opts = {})
+      super(servers, opts)
+    end
+
+    # Copy from https://github.com/crate/crate_ruby/blob/master/lib/crate_ruby/client.rb#L95
+    #
+    # Executes a SQL statement against the Crate HTTP REST endpoint.
+    # @param [String] sql statement to execute
+    # @param [Array] args Array of values used for parameter substitution
+    # @param [Hash] Net::HTTP options (open_timeout, read_timeout)
+    # @return [ResultSet]
+    def execute(sql, args = nil, bulk_args = nil, http_options = {})
+      #@logger.debug sql
+      req = Net::HTTP::Post.new("/_sql", initheader = {'Content-Type' => 'application/json'})
+      body = {"stmt" => sql}
+      body.merge!({'args' => args}) if args
+      body.merge!({'bulk_args' => bulk_args}) if bulk_args
+      req.body = body.to_json
+      response = request(req, http_options)
+      #@logger.debug response.body
+      success = case response.code
+                  when /^2\d{2}/
+                    ResultSet.new response.body
+                  else
+                    @logger.info(response.body)
+                    raise CrateRuby::CrateError.new(response.body)
+                end
+      success
+    end
+  end
+end

metadata

@@ -0,0 +1,95 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-cratedb
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- buom
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-12-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: crate_ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.8
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.8
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.12.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.12.0
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Fluent Output Plugin for CrateDB (http://crate.io)
+email:
+- me@buom.io
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- VERSION
+- fluent-plugin-cratedb.gemspec
+- lib/fluent/plugin/out_cratedb.rb
+homepage: https://github.com/buom/fluent-plugin-cratedb
+licenses:
+- Apache-2.0
+metadata: {}
+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.4.8
+signing_key: 
+specification_version: 4
+summary: A plugin for Fluentd
+test_files: []