fluent-plugin-ruby-kafka 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: a55057020e8d28616feca8a5de68b50dd43e3d7a
+  data.tar.gz: dd8ab1b9755587933c9b46c18729c0d50db7a442
+SHA512:
+  metadata.gz: b324b25cfa6b0e11082226752fe0b05a033a50d4dd8b48f6b5a27874e6b4d3214b599078af278a14d459fe522d4e21e3a83e9c3c50015242bc280019545abe07
+  data.tar.gz: 1a9c77c893306c675a2f2ef04590e0894db9d49de3120bce7192eeb2a340acf6d5967783bb57f186aced0054486ae3da688356413da4491a8b4286cefd7750fb

data/.gitignore

@@ -0,0 +1,17 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+*.bundle
+*.so
+*.o
+*.a
+*.gem
+mkmf.log
+.rbenv-gemsets
+.ruby-version

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,13 @@
+Copyright (c) 2015- Yoshimitsu KOKUBO
+
+   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,35 @@
+# fluent-plugin-ruby-kafka
+
+[Fluentd](http://fluentd.org/) output plugin to write data into kafka topic.
+
+It has been implemented in the reference [fluent-plugin-kafka](https://github.com/htgc/fluent-plugin-kafka)
+
+The difference between fluent-plugin-kafka are using [ruby-kafka](https://github.com/zendesk/ruby-kafka).
+
+## Configuration
+
+### RubyKafkaOutput
+
+Use zookeeper:
+
+    <match tag.**>
+      type ruby_kafka
+      zookeepers zookeeper01:2181,zookeeper02:2181
+      default_topic topic1
+    </match>
+
+Setting the brokers:
+
+    <match tag.**>
+      type ruby_kafka
+      brokers broker01:9092,broker02:9092
+      default_topic topic1
+    </match>
+
+## TODO
+
+## Copyright
+
+* Copyright (c) 2016- Yoshimitsu KOKUBO (yoshi)
+* License
+  * Apache License, Version 2.0

data/Rakefile

@@ -0,0 +1,11 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+task :default => :test

data/fluent-plugin-ruby-kafka.gemspec

@@ -0,0 +1,23 @@
+# coding: utf-8
+
+Gem::Specification.new do |gem|
+  gem.name          = "fluent-plugin-ruby-kafka"
+  gem.version       = "0.1.0"
+  gem.authors       = ["Yoshimitsu Kokubo"]
+  gem.email         = ["yoshi@unplus.net"]
+  gem.summary       = %q{Kafka's produce fluentd plugin by ruby-kafka}
+  gem.description   = %q{Kafka's produce fluentd plugin by ruby-kafka}
+  gem.homepage      = "https://github.com/unplus/fluent-plugin-ruby-kafka"
+  gem.license       = "Apache-2.0"
+
+  gem.files         = `git ls-files -z`.split("\x0")
+  gem.executables   = gem.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+
+  gem.add_development_dependency "rake"
+  gem.add_development_dependency "test-unit"
+  gem.add_dependency "fluentd", '>= 0.10.59'
+  gem.add_dependency "ruby-kafka", '~> 0.3.0', '>= 0.3.2'
+  gem.add_dependency "zookeeper", '~> 1.4.0', '>= 1.4.11'
+end

data/lib/fluent/plugin/out_ruby_kafka.rb

@@ -0,0 +1,96 @@
+# coding: utf-8
+
+class RubyKafkaOutput < Fluent::Output
+  Fluent::Plugin.register_output('ruby_kafka', self)
+
+  config_param :zookeepers, :string, :default => nil
+
+  config_param :brokers, :string, :default => nil
+
+  config_param :default_topic, :string, :default => nil
+
+  config_param :max_queue_size, :integer, :default => 1000
+
+  config_param :delivery_threshold, :integer, :default => 0
+
+  config_param :delivery_interval, :integer, :default => 0
+
+  config_param :retry_count, :integer, :default => 3
+
+  config_param :output_include_tag, :bool, :default => false
+
+  config_param :output_include_time, :bool, :default => false
+
+  config_param :use_kafka_log, :bool, :default => false
+
+  @seed_brokers = []
+
+  def initialize
+    super
+    require "kafka"
+    require "zookeeper"
+    require "json"
+  end
+
+  def configure(conf)
+    super
+    @seed_brokers = []
+    if @zookeepers
+      z = Zookeeper.new(@zookeepers)
+      z.get_children(:path => '/brokers/ids')[:children].each do |id|
+        broker = Yajl.load(z.get(:path => "/brokers/ids/#{id}")[:data])
+        @seed_brokers.push("#{broker['host']}:#{broker['port']}")
+      end
+      z.close
+      log.info "brokers has been refreshed via Zookeeper: #{@seed_brokers}"
+    end
+
+    if @seed_brokers.empty? and @brokers
+      @seed_brokers = @brokers.match(",").nil? ? [@brokers] : @brokers.split(",")
+      log.info "brokers has been set directly: #{@seed_brokers}"
+    end
+
+    raise Fluent::ConfigError, "Broker has not been set." if @seed_brokers.empty?
+
+    kafka = Kafka.new(seed_brokers: @seed_brokers, logger: @use_kafka_log ? log : nil)
+    @producer = kafka.async_producer(
+      max_queue_size: @max_queue_size,
+      delivery_threshold: @delivery_threshold,
+      delivery_interval: @delivery_interval,
+    )
+  end
+
+  def start
+    super
+  end
+
+  def shutdown
+    super
+    @producer.shutdown if @producer
+  end
+
+  def emit(tag, es, chain)
+    chain.next
+    es.each do |time,record|
+      record['time'] = time if @output_include_time
+      record['tag'] = tag if @output_include_tag
+      topic = record['topic'] || @default_topic || tag
+
+      data = JSON.dump(record)
+      retry_counter = 0
+      begin
+        retry_counter += 1
+        @producer.produce(data, topic: topic)
+      rescue Kafka::BufferOverflow
+        if retry_counter <= @retry_count
+          log.warn "Buffer overflow, backing off for 1s. #{retry_counter} time."
+          sleep 1
+          retry
+        else
+          raise
+        end
+      end
+    end
+  end
+
+end

data/test/helper.rb

@@ -0,0 +1,32 @@
+# coding: utf-8
+
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+require 'fluent/test'
+
+unless ENV.has_key?('VERBOSE')
+  nulllogger = Object.new
+  nulllogger.instance_eval {|obj|
+    def method_missing(method, *args)
+      # pass
+    end
+  }
+  $log = nulllogger
+end
+
+require 'fluent/plugin/out_ruby_kafka'
+
+class Test::Unit::TestCase
+end

data/test/plugin/test_out_ruby_kafka.rb

@@ -0,0 +1,24 @@
+require 'helper'
+
+class RubyKafkaOutputTest < Test::Unit::TestCase
+
+  def setup
+    Fluent::Test.setup
+  end
+
+  CONFIG = %[
+      zookeepers zookeeper01:2181,zookeeper02:2181
+      default_topic topic1
+      max_queue_size 1000
+      delivery_threshold 0
+      delivery_interval 0
+      output_include_tag  true
+      output_include_time true
+      log_level info
+  ]
+
+  def create_driver(conf = CONFIG, tag='test')
+    Fluent::Test::OutputTestDriver.new(RubyKafkaOutput, tag).configure(conf)
+  end
+
+end

metadata

@@ -0,0 +1,137 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-ruby-kafka
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Yoshimitsu Kokubo
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-04-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.10.59
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.10.59
+- !ruby/object:Gem::Dependency
+  name: ruby-kafka
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.0
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.3.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.0
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.3.2
+- !ruby/object:Gem::Dependency
+  name: zookeeper
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.4.0
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.4.11
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.4.0
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.4.11
+description: Kafka's produce fluentd plugin by ruby-kafka
+email:
+- yoshi@unplus.net
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- fluent-plugin-ruby-kafka.gemspec
+- lib/fluent/plugin/out_ruby_kafka.rb
+- test/helper.rb
+- test/plugin/test_out_ruby_kafka.rb
+homepage: https://github.com/unplus/fluent-plugin-ruby-kafka
+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.2.2
+signing_key: 
+specification_version: 4
+summary: Kafka's produce fluentd plugin by ruby-kafka
+test_files:
+- test/helper.rb
+- test/plugin/test_out_ruby_kafka.rb