fluent-plugin-copy_ex 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e55e9309b56820ce9c6e3889d78a7adef469a666
+  data.tar.gz: 819fd35752ace0455f78083abfb998fa6dbd0a78
+SHA512:
+  metadata.gz: afb6a9e80ba971b9a2b7916f797f1a15e82e8c45da068819e6af2badd5dc7cfed789b587444e2eac9f7d275919aeef3890e5c201d11d93311d640cb577c0c285
+  data.tar.gz: f2aefd89cbb13156b6b14215f8e3b32a7f6f85e406c8d71a508f98f0f1e3f223a85e64f7065f3c5d25608799f684b7b6fb5e21790ad3c2c29428e85c7241261d

data/.gitignore

@@ -0,0 +1,13 @@
+/*.gem
+~*
+#*
+*~
+.bundle
+Gemfile.lock
+.rbenv-version
+vendor
+doc/*
+tmp/*
+.yardoc
+.ruby-version
+pkg/*

data/.travis.yml

@@ -0,0 +1,6 @@
+rvm:
+  - 1.9.3
+  - 2.0.0
+  - 2.1.*
+gemfile:
+  - Gemfile

data/CHANGELOG.md

@@ -0,0 +1,4 @@
+## 0.0.1
+
+First version
+

data/Gemfile

@@ -0,0 +1,3 @@
+source "http://rubygems.org"
+
+gemspec

data/LICENSE

@@ -0,0 +1 @@
+Same with Fluentd

data/README.md

@@ -0,0 +1,63 @@
+# fluent-plugin-copy_ex
+
+## About
+
+Fluentd out\_copy extension
+
+## What is this for?
+
+Think of `out_copy` configuration as folloings:
+
+```apache
+<match **>
+  type copy
+  <store>
+    type plugin1
+  </store>
+  <store>
+    type plugin2
+  </store>
+</match>
+```
+
+In the current Fluentd, when plugin1 raises an error internally, the chain is broken and the plugin2 is not executed. 
+
+The `out_copy_ex` supplies `ignore_error` option so that it will not break the chain and the plugin2 is executed. 
+
+See https://github.com/fluent/fluentd/pull/303 for discussions. 
+
+
+## Configuration
+
+```apache
+<match **>
+  type copy_ex
+  <store ignore_error>
+    type plugin1
+  </store>
+  <store ignore_error>
+    type plugin2
+  </store>
+</match>
+```
+
+## Parameters
+
+Basically same with out\_copy plugin. See http://docs.fluentd.org/articles/out_copy
+    
+## 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](../../pull/new/master)
+
+## ChangeLog
+
+See [CHANGELOG.md](CHANGELOG.md) for details.
+
+## Copyright
+
+* Copyright (c) 2014- Naotoshi Seo
+* See [LICENSE](LICENSE) for details.

data/Rakefile

@@ -0,0 +1,16 @@
+# encoding: utf-8
+require "bundler/gem_tasks"
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/*.rb'
+  test.verbose = true
+end
+task :default => :test
+
+desc 'Open an irb session preloaded with the gem library'
+task :console do
+    sh 'irb -rubygems -I lib'
+end
+task :c => :console

data/example/fluent.conf

@@ -0,0 +1,9 @@
+<match **>
+  type copy_ex
+  <store ignore-failure>
+     type stdout
+  </store>
+  <store ignore-failure>
+     type stdout
+  </store>
+</match>

data/fluent-plugin-copy_ex.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "fluent-plugin-copy_ex"
+  s.version     = "0.0.1"
+  s.authors     = ["Naotoshi Seo"]
+  s.email       = ["sonots@gmail.com"]
+  s.homepage    = "https://github.com/sonots/fluent-plugin-copy_ex"
+  s.summary     = "Fluentd out_copy extension"
+  s.description = s.summary
+  s.licenses    = ["MIT"]
+
+  s.rubyforge_project = "fluent-plugin-copy_ex"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_runtime_dependency "fluentd"
+  s.add_development_dependency "rake"
+  s.add_development_dependency "pry"
+  s.add_development_dependency "pry-nav"
+end

data/lib/fluent/plugin/out_copy_ex.rb

@@ -0,0 +1,72 @@
+module Fluent
+  class CopyExOutput < MultiOutput
+    Plugin.register_output('copy_ex', self)
+
+    config_param :deep_copy, :bool, :default => false
+
+    def initialize
+      super
+      @outputs = []
+      @ignore_errors = []
+    end
+
+    attr_reader :outputs, :ignore_errors
+
+    def configure(conf)
+      super
+      conf.elements.select {|e|
+        e.name == 'store'
+      }.each {|e|
+        type = e['type']
+        unless type
+          raise ConfigError, "Missing 'type' parameter on <store> directive"
+        end
+        log.debug "adding store type=#{type.dump}"
+
+        output = Plugin.new_output(type)
+        output.configure(e)
+        @outputs << output
+
+        @ignore_errors << (e.arg == "ignore_error")
+      }
+    end
+
+    def start
+      @outputs.each {|o|
+        o.start
+      }
+    end
+
+    def shutdown
+      @outputs.each {|o|
+        o.shutdown
+      }
+    end
+
+    def emit(tag, es, chain)
+      unless es.repeatable?
+        m = MultiEventStream.new
+        es.each {|time,record|
+          m.add(time, record)
+        }
+        es = m
+      end
+
+      # Here, we do not use OutputChain for custom
+      @outputs.each_index do |idx|
+        _es = @deep_copy ? es.dup : es
+        begin
+          @outputs[idx].emit(tag, _es, NullOutputChain.instance)
+        rescue => e
+          if @ignore_errors[idx]
+            log.error :error_class => e.class, :error => e.message
+          else
+            raise e
+          end
+        end
+      end
+
+      chain.next
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1,18 @@
+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'
+require 'fluent/plugin/out_copy_ex'
+
+class Test::Unit::TestCase
+end

data/test/plugin/test_out_copy_ex.rb

@@ -0,0 +1,208 @@
+require 'fluent/test'
+
+class CopyExOutputTest < Test::Unit::TestCase
+  def setup
+    Fluent::Test.setup
+  end
+
+  CONFIG = %[
+    <store>
+      type test
+      name c0
+    </store>
+    <store>
+      type test
+      name c1
+    </store>
+    <store>
+      type test
+      name c2
+    </store>
+  ]
+
+  IGNORE_ERROR_CONFIG = %[
+    <store ignore_error>
+      type test
+      name c0
+    </store>
+    <store ignore_error>
+      type test
+      name c1
+    </store>
+    <store>
+      type test
+      name c2
+    </store>
+  ]
+
+  def create_driver(conf = CONFIG)
+    Fluent::Test::OutputTestDriver.new(Fluent::CopyExOutput).configure(conf)
+  end
+
+  def test_configure
+    d = create_driver
+
+    outputs = d.instance.outputs
+    assert_equal 3, outputs.size
+    assert_equal Fluent::TestOutput, outputs[0].class
+    assert_equal Fluent::TestOutput, outputs[1].class
+    assert_equal Fluent::TestOutput, outputs[2].class
+    assert_equal "c0", outputs[0].name
+    assert_equal "c1", outputs[1].name
+    assert_equal "c2", outputs[2].name
+  end
+
+  def test_configure_ignore_error
+    d = create_driver(IGNORE_ERROR_CONFIG)
+
+    outputs = d.instance.outputs
+    ignore_errors = d.instance.ignore_errors
+    assert_equal outputs.size, ignore_errors.size
+    assert_equal true,  ignore_errors[0]
+    assert_equal true,  ignore_errors[1]
+    assert_equal false, ignore_errors[2]
+  end
+
+  def test_emit
+    d = create_driver
+
+    time = Time.parse("2011-01-02 13:14:15 UTC").to_i
+    d.emit({"a"=>1}, time)
+    d.emit({"a"=>2}, time)
+
+    d.instance.outputs.each {|o|
+      assert_equal [
+          [time, {"a"=>1}],
+          [time, {"a"=>2}],
+        ], o.events
+    }
+  end
+
+  def test_msgpack_es_emit_bug
+    d = Fluent::Test::OutputTestDriver.new(Fluent::CopyExOutput)
+
+    outputs = %w(p1 p2).map do |pname|
+      p = Fluent::Plugin.new_output('test')
+      p.configure('name' => pname)
+      p.define_singleton_method(:emit) do |tag, es, chain|
+        es.each do |time, record|
+          super(tag, [[time, record]], chain)
+        end
+      end
+      p
+    end
+
+    d.instance.instance_eval { @outputs = outputs }
+
+    es = if defined?(MessagePack::Packer)
+           time = Time.parse("2013-05-26 06:37:22 UTC").to_i
+           packer = MessagePack::Packer.new
+           packer.pack([time, {"a" => 1}])
+           packer.pack([time, {"a" => 2}])
+           Fluent::MessagePackEventStream.new(packer.to_s)
+         else
+           events = "#{[time, {"a" => 1}].to_msgpack}#{[time, {"a" => 2}].to_msgpack}"
+           Fluent::MessagePackEventStream.new(events)
+         end
+
+    d.instance.emit('test', es, Fluent::NullOutputChain.instance)
+
+    d.instance.outputs.each { |o|
+      assert_equal [
+        [time, {"a"=>1}],
+        [time, {"a"=>2}],
+      ], o.events
+    }
+  end
+
+  def create_event_test_driver(is_deep_copy = false)
+    deep_copy_config = %[
+deep_copy true
+]
+
+    output1 = Fluent::Plugin.new_output('test')
+    output1.configure('name' => 'output1')
+    output1.define_singleton_method(:emit) do |tag, es, chain|
+      es.each do |time, record|
+        record['foo'] = 'bar'
+        super(tag, [[time, record]], chain)
+      end
+    end
+
+    output2 = Fluent::Plugin.new_output('test')
+    output2.configure('name' => 'output2')
+    output2.define_singleton_method(:emit) do |tag, es, chain|
+      es.each do |time, record|
+        super(tag, [[time, record]], chain)
+      end
+    end
+
+    outputs = [output1, output2]
+
+    d = Fluent::Test::OutputTestDriver.new(Fluent::CopyExOutput)
+    d = d.configure(deep_copy_config) if is_deep_copy
+    d.instance.instance_eval { @outputs = outputs }
+    d
+  end
+
+  def test_one_event
+    time = Time.parse("2013-05-26 06:37:22 UTC").to_i
+
+    d = create_event_test_driver(false)
+    es = Fluent::OneEventStream.new(time, {"a" => 1})
+    d.instance.emit('test', es, Fluent::NullOutputChain.instance)
+
+    assert_equal [
+      [[time, {"a"=>1, "foo"=>"bar"}]],
+      [[time, {"a"=>1, "foo"=>"bar"}]]
+    ], d.instance.outputs.map{ |o| o.events }
+
+    d = create_event_test_driver(true)
+    es = Fluent::OneEventStream.new(time, {"a" => 1})
+    d.instance.emit('test', es, Fluent::NullOutputChain.instance)
+
+    assert_equal [
+      [[time, {"a"=>1, "foo"=>"bar"}]],
+      [[time, {"a"=>1}]]
+    ], d.instance.outputs.map{ |o| o.events }
+  end
+
+  def test_multi_event
+    time = Time.parse("2013-05-26 06:37:22 UTC").to_i
+
+    d = create_event_test_driver(false)
+    es = Fluent::MultiEventStream.new
+    es.add(time, {"a" => 1})
+    es.add(time, {"b" => 2})
+    d.instance.emit('test', es, Fluent::NullOutputChain.instance)
+
+    assert_equal [
+      [[time, {"a"=>1, "foo"=>"bar"}], [time, {"b"=>2, "foo"=>"bar"}]],
+      [[time, {"a"=>1, "foo"=>"bar"}], [time, {"b"=>2, "foo"=>"bar"}]]
+    ], d.instance.outputs.map{ |o| o.events }
+
+    d = create_event_test_driver(true)
+    es = Fluent::MultiEventStream.new
+    es.add(time, {"a" => 1})
+    es.add(time, {"b" => 2})
+    d.instance.emit('test', es, Fluent::NullOutputChain.instance)
+
+    assert_equal [
+      [[time, {"a"=>1, "foo"=>"bar"}], [time, {"b"=>2, "foo"=>"bar"}]],
+      [[time, {"a"=>1}], [time, {"b"=>2}]]
+    ], d.instance.outputs.map{ |o| o.events }
+  end
+
+  def test_ignore_error
+    d = create_driver(IGNORE_ERROR_CONFIG)
+
+    # override to raise an error
+    d.instance.outputs.first.define_singleton_method(:emit) do |tag, es, chain|
+      raise ArgumentError, 'Failed'
+    end
+
+    time = Time.parse("2011-01-02 13:14:15 UTC").to_i
+    assert_nothing_raised { d.emit({"a"=>1}, time) }
+  end
+end
+

metadata

@@ -0,0 +1,114 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-copy_ex
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Naotoshi Seo
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-04-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !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: pry
+  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: pry-nav
+  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'
+description: Fluentd out_copy extension
+email:
+- sonots@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- CHANGELOG.md
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- example/fluent.conf
+- fluent-plugin-copy_ex.gemspec
+- lib/fluent/plugin/out_copy_ex.rb
+- test/helper.rb
+- test/plugin/test_out_copy_ex.rb
+homepage: https://github.com/sonots/fluent-plugin-copy_ex
+licenses:
+- MIT
+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: fluent-plugin-copy_ex
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Fluentd out_copy extension
+test_files:
+- test/helper.rb
+- test/plugin/test_out_copy_ex.rb