fluent-plugin-unit-time-filter 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: cb21bb8de5296e1949d37a65ae7edb5e7ac6cccb
+  data.tar.gz: 1ac61305e86d61a953d9c56756d9731f596f4cbc
+SHA512:
+  metadata.gz: e1eab0e7c3d5f0c351a73ef7756dcef08837c38c8e8d5edd5127324e6ca0733e541b3343966bcd77ec2c352bd2c4b5955384f6c893c8bb1b5a7ff24a3504bd37
+  data.tar.gz: f89b7f1b131bdb410ec6908864b448fd6b9ea0d8c4d4e14c67d06d2cabf1bc97f8ccfff21ca767dde0a87a6f2837902f13e2d3f89516d43efa61fd56b0ace073

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.rspec

@@ -0,0 +1,2 @@
+--require spec_helper
+--colour

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Genki Sugawara
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,73 @@
+# fluent-plugin-unit-time-filter
+
+Plug-in to aggregate by unit time
+
+[![Gem Version](https://badge.fury.io/rb/fluent-plugin-unit-time-filter.png)](http://badge.fury.io/rb/fluent-plugin-unit-time-filter)
+[![Build Status](https://drone.io/bitbucket.org/winebarrel/fluent-plugin-unit-time-filter/status.png)](https://drone.io/bitbucket.org/winebarrel/fluent-plugin-unit-time-filter/latest)
+
+## Configuration
+
+```
+<match my.**>
+  type unit_time_filter
+  filter_path /foo/bar/my_filrer.rb
+  unit_sec 10
+  #prefix filtered
+  #emit_each_tag false
+  #pass_hash_row false
+  #hash_row_time_key time
+  #hash_row_tag_key tag
+</match>
+
+<match filtered.my.**>
+  type stdout
+</match>
+```
+
+## Filter
+
+```ruby
+# Count the number of records of every unit time
+
+proc {|records|
+
+  # `records` is an Array such as:
+  #   [[tag, time, record], [tag, time, record], ...]
+  #
+  # e.g.)
+  #   [["my.data", 1391820170, {"hoge"=>"fuga"}],
+  #    ["my.data", 1391820170, {"hoge"=>"fuga"}],
+  #    ["my.data", 1391820170, {"hoge"=>"fuga"}],
+  #    ...
+
+  {'count' => records.count}
+  # or [{...},{...},{...}, ...]
+}
+```
+
+## Usage
+
+Run the following command:
+
+```sh
+while true; do
+  echo '{"hoge":"fuga"}' | fluent-cat my.data
+done
+```
+
+fluentd outputs the following:
+
+```
+2014-02-08 00:38:40 +0900 filtered.my.data: {"count":42}
+2014-02-08 00:38:50 +0900 filtered.my.data: {"count":43}
+2014-02-08 00:39:00 +0900 filtered.my.data: {"count":41}
+...
+```
+
+## 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

data/Rakefile

@@ -0,0 +1,5 @@
+require "bundler/gem_tasks"
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new('spec')
+task :default => :spec

data/fluent-plugin-unit-time-filter.gemspec

@@ -0,0 +1,21 @@
+# coding: utf-8
+Gem::Specification.new do |spec|
+  spec.name          = "fluent-plugin-unit-time-filter"
+  spec.version       = "0.1.0"
+  spec.authors       = ["Genki Sugawara"]
+  spec.email         = ["sgwr_dts@yahoo.co.jp"]
+  spec.description   = %q{Plug-in to aggregate by unit time}
+  spec.summary       = %q{Plug-in to aggregate by unit time}
+  spec.homepage      = "https://bitbucket.org/winebarrel/fluent-plugin-unit-time-filter"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  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 "fluentd"
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency 'rspec', '>= 2.11.0'
+end

data/lib/fluent/plugin/out_unit_time_filter.rb

@@ -0,0 +1,63 @@
+require 'fluent/plugin/unit_time_filter_buffer'
+
+class Fluent::UnitTimeFilterOutput < Fluent::Output
+  Fluent::Plugin.register_output('unit_time_filter', self)
+
+  config_param :filter_path,       :type => :string
+  config_param :unit_sec,          :type => :size,   :default => 1
+  config_param :prefix,            :type => :string, :default => 'filtered'
+  config_param :emit_each_tag,     :type => :bool,   :default => false
+  config_param :pass_hash_row,     :type => :bool,   :default => false
+  config_param :hash_row_time_key, :type => :string, :default => 'time'
+  config_param :hash_row_tag_key,  :type => :string, :default => 'tag'
+
+  BUFFER_KEY = :unit_time_filter_buffer
+
+  def configure(conf)
+    super
+
+    unless File.exist?(@filter_path)
+      raise Fluent::ConfigError, "No such file: #{@filter_path}"
+    end
+
+    begin
+      @filter = Object.new.instance_eval(File.read(@filter_path))
+    rescue => e
+      raise Fluent::ConfigError, "Invalid filter: #{@filter_path}: #{e}"
+    end
+
+    unless @filter.kind_of?(Proc)
+      raise Fluent::ConfigError, "Invalid filter: #{@filter_path}: Filter must be Proc"
+    end
+  end
+
+  def shutdown
+    super
+    Thread.current[BUFFER_KEY] = nil
+  end
+
+  def buffer
+    buf = Thread.current[BUFFER_KEY]
+
+    unless buf
+      buf = Buffer.new({
+        :filter            => @filter,
+        :unit_sec          => @unit_sec,
+        :prefix            => @prefix,
+        :emit_each_tag     => @emit_each_tag,
+        :pass_hash_row     => @pass_hash_row,
+        :hash_row_time_key => @hash_row_time_key,
+        :hash_row_tag_key  => @hash_row_tag_key,
+      })
+
+      Thread.current[BUFFER_KEY] = buf
+    end
+
+    return buf
+  end
+
+  def emit(tag, es, chain)
+    buffer.resume(tag, es)
+    chain.next
+  end
+end

data/lib/fluent/plugin/unit_time_filter_buffer.rb

@@ -0,0 +1,99 @@
+class Fluent::UnitTimeFilterOutput < Fluent::Output
+  class Buffer
+    include Enumerable
+
+    def initialize(options)
+      @init_queue = []
+
+      @fiber = Fiber.new do |tag, es|
+        @init_queue << [tag, es]
+        run(options)
+      end
+    end
+
+    def each
+      loop do
+        tag, es = @init_queue.shift || Fiber.yield
+        break unless (tag && es)
+
+        es.each do |time, record|
+          yield([tag, time, record])
+        end
+      end
+    end
+
+    def resume(tag, es)
+      @fiber.resume(tag, es)
+    end
+
+    private
+
+    def run(options)
+      filter        = options[:filter]
+      unit_sec      = options[:unit_sec]
+      prefix        = options[:prefix]
+      emit_each_tag = options[:emit_each_tag]
+      pass_hash_row = options[:pass_hash_row]
+
+      prev_time = nil
+
+      self.chunk {|tag, time, record|
+        time
+      }.slice_before {|time, records|
+        result = (time % unit_sec).zero? || (prev_time && minus_r(time, unit_sec) != minus_r(prev_time, unit_sec))
+        prev_time = time
+        result
+      }.each {|records|
+        records = records.inject([]) {|r, i| r + i[1] }
+        records = conv_to_hash_rows_if_needed(records, options)
+
+        time = get_time(records.first, options)
+        time -= time % unit_sec
+
+        rs = filter.call(records)
+        rs = [rs] unless rs.kind_of?(Array)
+
+        if emit_each_tag
+          tags = records.map {|r| get_tag(r, options) }.uniq
+          tags.each {|t| emit_records(prefix, t, time, rs) }
+        else
+          tag = get_tag(records.first, options)
+          emit_records(prefix, tag, time, rs)
+        end
+      }
+    end
+
+    def emit_records(prefix, tag, time, records)
+      records.each do |record|
+        Fluent::Engine.emit("#{prefix}.#{tag}", time, record)
+      end
+    end
+
+    def conv_to_hash_rows_if_needed(records, options)
+      return records unless options[:pass_hash_row]
+
+      time_key = options[:hash_row_time_key]
+      tag_key  = options[:hash_row_tag_key]
+
+      records.map do |tag, time, r|
+        r[time_key] = time
+        r[tag_key]  = tag
+        r
+      end
+    end
+
+    def get_time(record, options)
+      time_key = options[:hash_row_time_key]
+      options[:pass_hash_row] ? record[time_key] : record[1]
+    end
+
+    def get_tag(record, options)
+      tag_key = options[:hash_row_tag_key]
+      options[:pass_hash_row] ? record[tag_key] : record[0]
+    end
+
+    def minus_r(m, n)
+      m - (m % n)
+    end
+  end
+end

data/spec/out_unit_time_filter_spec.rb

@@ -0,0 +1,236 @@
+describe Fluent::UnitTimeFilterOutput do
+  let(:time) {
+    Time.parse('2014-02-08 13:14:15 +0900').to_i
+  }
+
+  describe 'when 2 records transmitted per second' do
+    it 'should be aggregated to two records' do
+      run_driver do |d|
+        (0...10).each do |i|
+          d.emit({"key#{i}" => "val#{i}"}, time + i)
+          d.emit({"key#{i}_" => "val#{i}_"}, time + i)
+        end
+
+        expect(d.emits).to eq(
+          [["filtered.test.default", 1391832855, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832855, {\"key0\"=>\"val0\"}], [\"test.default\", 1391832855, {\"key0_\"=>\"val0_\"}]]"}],
+           ["filtered.test.default", 1391832856, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832856, {\"key1\"=>\"val1\"}], [\"test.default\", 1391832856, {\"key1_\"=>\"val1_\"}]]"}],
+           ["filtered.test.default", 1391832857, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832857, {\"key2\"=>\"val2\"}], [\"test.default\", 1391832857, {\"key2_\"=>\"val2_\"}]]"}],
+           ["filtered.test.default", 1391832858, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832858, {\"key3\"=>\"val3\"}], [\"test.default\", 1391832858, {\"key3_\"=>\"val3_\"}]]"}],
+           ["filtered.test.default", 1391832859, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832859, {\"key4\"=>\"val4\"}], [\"test.default\", 1391832859, {\"key4_\"=>\"val4_\"}]]"}],
+           ["filtered.test.default", 1391832860, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832860, {\"key5\"=>\"val5\"}], [\"test.default\", 1391832860, {\"key5_\"=>\"val5_\"}]]"}],
+           ["filtered.test.default", 1391832861, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832861, {\"key6\"=>\"val6\"}], [\"test.default\", 1391832861, {\"key6_\"=>\"val6_\"}]]"}],
+           ["filtered.test.default", 1391832862, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832862, {\"key7\"=>\"val7\"}], [\"test.default\", 1391832862, {\"key7_\"=>\"val7_\"}]]"}]]
+        )
+      end
+    end
+
+    it 'should be rewritten to the specified tag' do
+      run_driver(:prefix => 'any_prefix') do |d|
+        (0...10).each do |i|
+          d.emit({"key#{i}" => "val#{i}"}, time + i)
+          d.emit({"key#{i}_" => "val#{i}_"}, time + i)
+        end
+
+        expect(d.emits).to eq(
+          [["any_prefix.test.default", 1391832855, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832855, {\"key0\"=>\"val0\"}], [\"test.default\", 1391832855, {\"key0_\"=>\"val0_\"}]]"}],
+           ["any_prefix.test.default", 1391832856, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832856, {\"key1\"=>\"val1\"}], [\"test.default\", 1391832856, {\"key1_\"=>\"val1_\"}]]"}],
+           ["any_prefix.test.default", 1391832857, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832857, {\"key2\"=>\"val2\"}], [\"test.default\", 1391832857, {\"key2_\"=>\"val2_\"}]]"}],
+           ["any_prefix.test.default", 1391832858, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832858, {\"key3\"=>\"val3\"}], [\"test.default\", 1391832858, {\"key3_\"=>\"val3_\"}]]"}],
+           ["any_prefix.test.default", 1391832859, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832859, {\"key4\"=>\"val4\"}], [\"test.default\", 1391832859, {\"key4_\"=>\"val4_\"}]]"}],
+           ["any_prefix.test.default", 1391832860, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832860, {\"key5\"=>\"val5\"}], [\"test.default\", 1391832860, {\"key5_\"=>\"val5_\"}]]"}],
+           ["any_prefix.test.default", 1391832861, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832861, {\"key6\"=>\"val6\"}], [\"test.default\", 1391832861, {\"key6_\"=>\"val6_\"}]]"}],
+           ["any_prefix.test.default", 1391832862, {"count"=>2, "inspect"=>"[[\"test.default\", 1391832862, {\"key7\"=>\"val7\"}], [\"test.default\", 1391832862, {\"key7_\"=>\"val7_\"}]]"}]]
+        )
+      end
+    end
+
+    it 'should receive multiple converted records' do
+      filter = <<-EOS
+        proc {|rs|
+          [
+            {'foo' => 'bar', 100 => 200},
+            {'bar' => 'baz', 200 => 300},
+          ]
+        }
+      EOS
+
+      run_driver(:filter => filter) do |d|
+        (0...10).each do |i|
+          d.emit({"key#{i}" => "val#{i}"}, time + i)
+          d.emit({"key#{i}_" => "val#{i}_"}, time + i)
+        end
+
+        expect(d.emits).to eq(
+          [["filtered.test.default", 1391832855, {"foo"=>"bar", 100=>200}],
+           ["filtered.test.default", 1391832855, {"bar"=>"baz", 200=>300}],
+           ["filtered.test.default", 1391832856, {"foo"=>"bar", 100=>200}],
+           ["filtered.test.default", 1391832856, {"bar"=>"baz", 200=>300}],
+           ["filtered.test.default", 1391832857, {"foo"=>"bar", 100=>200}],
+           ["filtered.test.default", 1391832857, {"bar"=>"baz", 200=>300}],
+           ["filtered.test.default", 1391832858, {"foo"=>"bar", 100=>200}],
+           ["filtered.test.default", 1391832858, {"bar"=>"baz", 200=>300}],
+           ["filtered.test.default", 1391832859, {"foo"=>"bar", 100=>200}],
+           ["filtered.test.default", 1391832859, {"bar"=>"baz", 200=>300}],
+           ["filtered.test.default", 1391832860, {"foo"=>"bar", 100=>200}],
+           ["filtered.test.default", 1391832860, {"bar"=>"baz", 200=>300}],
+           ["filtered.test.default", 1391832861, {"foo"=>"bar", 100=>200}],
+           ["filtered.test.default", 1391832861, {"bar"=>"baz", 200=>300}],
+           ["filtered.test.default", 1391832862, {"foo"=>"bar", 100=>200}],
+           ["filtered.test.default", 1391832862, {"bar"=>"baz", 200=>300}]]
+        )
+      end
+    end
+  end
+
+  describe 'when 3 records transmitted per second' do
+    it 'should be aggregated to two records' do
+      run_driver do |d|
+        (0...10).each do |i|
+          d.emit({"key#{i}" => "val#{i}"}, time + i)
+          d.emit({"key#{i}_" => "val#{i}_"}, time + i)
+          d.emit({"key#{i}__" => "val#{i}__"}, time + i)
+        end
+
+        expect(d.emits).to eq(
+          [["filtered.test.default", 1391832855, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832855, {\"key0\"=>\"val0\"}], [\"test.default\", 1391832855, {\"key0_\"=>\"val0_\"}], [\"test.default\", 1391832855, {\"key0__\"=>\"val0__\"}]]"}],
+           ["filtered.test.default", 1391832856, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832856, {\"key1\"=>\"val1\"}], [\"test.default\", 1391832856, {\"key1_\"=>\"val1_\"}], [\"test.default\", 1391832856, {\"key1__\"=>\"val1__\"}]]"}],
+           ["filtered.test.default", 1391832857, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832857, {\"key2\"=>\"val2\"}], [\"test.default\", 1391832857, {\"key2_\"=>\"val2_\"}], [\"test.default\", 1391832857, {\"key2__\"=>\"val2__\"}]]"}],
+           ["filtered.test.default", 1391832858, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832858, {\"key3\"=>\"val3\"}], [\"test.default\", 1391832858, {\"key3_\"=>\"val3_\"}], [\"test.default\", 1391832858, {\"key3__\"=>\"val3__\"}]]"}],
+           ["filtered.test.default", 1391832859, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832859, {\"key4\"=>\"val4\"}], [\"test.default\", 1391832859, {\"key4_\"=>\"val4_\"}], [\"test.default\", 1391832859, {\"key4__\"=>\"val4__\"}]]"}],
+           ["filtered.test.default", 1391832860, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832860, {\"key5\"=>\"val5\"}], [\"test.default\", 1391832860, {\"key5_\"=>\"val5_\"}], [\"test.default\", 1391832860, {\"key5__\"=>\"val5__\"}]]"}],
+           ["filtered.test.default", 1391832861, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832861, {\"key6\"=>\"val6\"}], [\"test.default\", 1391832861, {\"key6_\"=>\"val6_\"}], [\"test.default\", 1391832861, {\"key6__\"=>\"val6__\"}]]"}],
+           ["filtered.test.default", 1391832862, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832862, {\"key7\"=>\"val7\"}], [\"test.default\", 1391832862, {\"key7_\"=>\"val7_\"}], [\"test.default\", 1391832862, {\"key7__\"=>\"val7__\"}]]"}]]
+        )
+      end
+    end
+  end
+
+  describe 'when unit_sec 10' do
+    it 'should be aggregated by every 10 sec' do
+      run_driver(:unit_sec => 10) do |d|
+        (0...30).step(5) do |i|
+          d.emit({"key#{i}" => "val#{i}"}, time + i)
+          d.emit({"key#{i}_" => "val#{i}_"}, time + i)
+          d.emit({"key#{i}__" => "val#{i}__"}, time + i)
+        end
+
+        expect(d.emits).to eq(
+          [["filtered.test.default", 1391832850, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832855, {\"key0\"=>\"val0\"}], [\"test.default\", 1391832855, {\"key0_\"=>\"val0_\"}], [\"test.default\", 1391832855, {\"key0__\"=>\"val0__\"}]]"}],
+           ["filtered.test.default", 1391832860, {"count"=>6, "inspect"=>"[[\"test.default\", 1391832860, {\"key5\"=>\"val5\"}], [\"test.default\", 1391832860, {\"key5_\"=>\"val5_\"}], [\"test.default\", 1391832860, {\"key5__\"=>\"val5__\"}], [\"test.default\", 1391832865, {\"key10\"=>\"val10\"}], [\"test.default\", 1391832865, {\"key10_\"=>\"val10_\"}], [\"test.default\", 1391832865, {\"key10__\"=>\"val10__\"}]]"}]]
+        )
+      end
+    end
+
+    it 'should be aggregated by every 10 sec' do
+      run_driver(:unit_sec => 10) do |d|
+        (0...100).step(20) do |i|
+          d.emit({"key#{i}" => "val#{i}"}, time + i)
+          d.emit({"key#{i}_" => "val#{i}_"}, time + i)
+          d.emit({"key#{i}__" => "val#{i}__"}, time + i)
+        end
+
+        expect(d.emits).to eq(
+          [["filtered.test.default", 1391832850, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832855, {\"key0\"=>\"val0\"}], [\"test.default\", 1391832855, {\"key0_\"=>\"val0_\"}], [\"test.default\", 1391832855, {\"key0__\"=>\"val0__\"}]]"}],
+           ["filtered.test.default", 1391832870, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832875, {\"key20\"=>\"val20\"}], [\"test.default\", 1391832875, {\"key20_\"=>\"val20_\"}], [\"test.default\", 1391832875, {\"key20__\"=>\"val20__\"}]]"}],
+           ["filtered.test.default", 1391832890, {"count"=>3, "inspect"=>"[[\"test.default\", 1391832895, {\"key40\"=>\"val40\"}], [\"test.default\", 1391832895, {\"key40_\"=>\"val40_\"}], [\"test.default\", 1391832895, {\"key40__\"=>\"val40__\"}]]"}]]
+        )
+      end
+    end
+  end
+
+  describe 'when multiple tags' do
+    it 'should be output only the first tag' do
+      run_driver do |d|
+        (0...10).each do |i|
+          d.tag = 'test.1'
+          d.emit({"key#{i}" => "val#{i}"}, time + i)
+          d.tag = 'test.2'
+          d.emit({"key#{i}_" => "val#{i}_"}, time + i)
+        end
+
+        expect(d.emits).to eq(
+          [["filtered.test.1", 1391832855, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832855, {\"key0\"=>\"val0\"}], [\"test.2\", 1391832855, {\"key0_\"=>\"val0_\"}]]"}],
+           ["filtered.test.1", 1391832856, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832856, {\"key1\"=>\"val1\"}], [\"test.2\", 1391832856, {\"key1_\"=>\"val1_\"}]]"}],
+           ["filtered.test.1", 1391832857, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832857, {\"key2\"=>\"val2\"}], [\"test.2\", 1391832857, {\"key2_\"=>\"val2_\"}]]"}],
+           ["filtered.test.1", 1391832858, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832858, {\"key3\"=>\"val3\"}], [\"test.2\", 1391832858, {\"key3_\"=>\"val3_\"}]]"}],
+           ["filtered.test.1", 1391832859, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832859, {\"key4\"=>\"val4\"}], [\"test.2\", 1391832859, {\"key4_\"=>\"val4_\"}]]"}],
+           ["filtered.test.1", 1391832860, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832860, {\"key5\"=>\"val5\"}], [\"test.2\", 1391832860, {\"key5_\"=>\"val5_\"}]]"}],
+           ["filtered.test.1", 1391832861, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832861, {\"key6\"=>\"val6\"}], [\"test.2\", 1391832861, {\"key6_\"=>\"val6_\"}]]"}],
+           ["filtered.test.1", 1391832862, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832862, {\"key7\"=>\"val7\"}], [\"test.2\", 1391832862, {\"key7_\"=>\"val7_\"}]]"}]]
+        )
+      end
+    end
+
+    it 'should be output to each tag' do
+      run_driver(:emit_each_tag => true) do |d|
+        (0...10).each do |i|
+          d.tag = 'test.1'
+          d.emit({"key#{i}" => "val#{i}"}, time + i)
+          d.tag = 'test.2'
+          d.emit({"key#{i}_" => "val#{i}_"}, time + i)
+        end
+
+        expect(d.emits).to eq(
+          [["filtered.test.1", 1391832855, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832855, {\"key0\"=>\"val0\"}], [\"test.2\", 1391832855, {\"key0_\"=>\"val0_\"}]]"}],
+           ["filtered.test.2", 1391832855, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832855, {\"key0\"=>\"val0\"}], [\"test.2\", 1391832855, {\"key0_\"=>\"val0_\"}]]"}],
+           ["filtered.test.1", 1391832856, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832856, {\"key1\"=>\"val1\"}], [\"test.2\", 1391832856, {\"key1_\"=>\"val1_\"}]]"}],
+           ["filtered.test.2", 1391832856, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832856, {\"key1\"=>\"val1\"}], [\"test.2\", 1391832856, {\"key1_\"=>\"val1_\"}]]"}],
+           ["filtered.test.1", 1391832857, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832857, {\"key2\"=>\"val2\"}], [\"test.2\", 1391832857, {\"key2_\"=>\"val2_\"}]]"}],
+           ["filtered.test.2", 1391832857, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832857, {\"key2\"=>\"val2\"}], [\"test.2\", 1391832857, {\"key2_\"=>\"val2_\"}]]"}],
+           ["filtered.test.1", 1391832858, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832858, {\"key3\"=>\"val3\"}], [\"test.2\", 1391832858, {\"key3_\"=>\"val3_\"}]]"}],
+           ["filtered.test.2", 1391832858, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832858, {\"key3\"=>\"val3\"}], [\"test.2\", 1391832858, {\"key3_\"=>\"val3_\"}]]"}],
+           ["filtered.test.1", 1391832859, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832859, {\"key4\"=>\"val4\"}], [\"test.2\", 1391832859, {\"key4_\"=>\"val4_\"}]]"}],
+           ["filtered.test.2", 1391832859, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832859, {\"key4\"=>\"val4\"}], [\"test.2\", 1391832859, {\"key4_\"=>\"val4_\"}]]"}],
+           ["filtered.test.1", 1391832860, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832860, {\"key5\"=>\"val5\"}], [\"test.2\", 1391832860, {\"key5_\"=>\"val5_\"}]]"}],
+           ["filtered.test.2", 1391832860, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832860, {\"key5\"=>\"val5\"}], [\"test.2\", 1391832860, {\"key5_\"=>\"val5_\"}]]"}],
+           ["filtered.test.1", 1391832861, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832861, {\"key6\"=>\"val6\"}], [\"test.2\", 1391832861, {\"key6_\"=>\"val6_\"}]]"}],
+           ["filtered.test.2", 1391832861, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832861, {\"key6\"=>\"val6\"}], [\"test.2\", 1391832861, {\"key6_\"=>\"val6_\"}]]"}],
+           ["filtered.test.1", 1391832862, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832862, {\"key7\"=>\"val7\"}], [\"test.2\", 1391832862, {\"key7_\"=>\"val7_\"}]]"}],
+           ["filtered.test.2", 1391832862, {"count"=>2, "inspect"=>"[[\"test.1\", 1391832862, {\"key7\"=>\"val7\"}], [\"test.2\", 1391832862, {\"key7_\"=>\"val7_\"}]]"}]]
+        )
+      end
+    end
+  end
+
+  describe 'when the array of Hash is passed' do
+    it 'should receive an array of Hash' do
+      run_driver(:pass_hash_row => true) do |d|
+        (0...10).each do |i|
+          d.emit({"key#{i}" => "val#{i}"}, time + i)
+          d.emit({"key#{i}_" => "val#{i}_"}, time + i)
+        end
+
+        expect(d.emits).to eq(
+          [["filtered.test.default", 1391832855, {"count"=>2, "inspect"=>"[{\"key0\"=>\"val0\", \"time\"=>1391832855, \"tag\"=>\"test.default\"}, {\"key0_\"=>\"val0_\", \"time\"=>1391832855, \"tag\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832856, {"count"=>2, "inspect"=>"[{\"key1\"=>\"val1\", \"time\"=>1391832856, \"tag\"=>\"test.default\"}, {\"key1_\"=>\"val1_\", \"time\"=>1391832856, \"tag\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832857, {"count"=>2, "inspect"=>"[{\"key2\"=>\"val2\", \"time\"=>1391832857, \"tag\"=>\"test.default\"}, {\"key2_\"=>\"val2_\", \"time\"=>1391832857, \"tag\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832858, {"count"=>2, "inspect"=>"[{\"key3\"=>\"val3\", \"time\"=>1391832858, \"tag\"=>\"test.default\"}, {\"key3_\"=>\"val3_\", \"time\"=>1391832858, \"tag\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832859, {"count"=>2, "inspect"=>"[{\"key4\"=>\"val4\", \"time\"=>1391832859, \"tag\"=>\"test.default\"}, {\"key4_\"=>\"val4_\", \"time\"=>1391832859, \"tag\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832860, {"count"=>2, "inspect"=>"[{\"key5\"=>\"val5\", \"time\"=>1391832860, \"tag\"=>\"test.default\"}, {\"key5_\"=>\"val5_\", \"time\"=>1391832860, \"tag\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832861, {"count"=>2, "inspect"=>"[{\"key6\"=>\"val6\", \"time\"=>1391832861, \"tag\"=>\"test.default\"}, {\"key6_\"=>\"val6_\", \"time\"=>1391832861, \"tag\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832862, {"count"=>2, "inspect"=>"[{\"key7\"=>\"val7\", \"time\"=>1391832862, \"tag\"=>\"test.default\"}, {\"key7_\"=>\"val7_\", \"time\"=>1391832862, \"tag\"=>\"test.default\"}]"}]]
+        )
+      end
+    end
+
+    it 'should receive an array of Hash (Not use the default key)' do
+      run_driver(:pass_hash_row => true, :hash_row_time_key => '__TIME__', :hash_row_tag_key => '__TAG__') do |d|
+        (0...10).each do |i|
+          d.emit({"key#{i}" => "val#{i}"}, time + i)
+          d.emit({"key#{i}_" => "val#{i}_"}, time + i)
+        end
+
+        expect(d.emits).to eq(
+          [["filtered.test.default", 1391832855, {"count"=>2, "inspect"=>"[{\"key0\"=>\"val0\", \"__TIME__\"=>1391832855, \"__TAG__\"=>\"test.default\"}, {\"key0_\"=>\"val0_\", \"__TIME__\"=>1391832855, \"__TAG__\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832856, {"count"=>2, "inspect"=>"[{\"key1\"=>\"val1\", \"__TIME__\"=>1391832856, \"__TAG__\"=>\"test.default\"}, {\"key1_\"=>\"val1_\", \"__TIME__\"=>1391832856, \"__TAG__\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832857, {"count"=>2, "inspect"=>"[{\"key2\"=>\"val2\", \"__TIME__\"=>1391832857, \"__TAG__\"=>\"test.default\"}, {\"key2_\"=>\"val2_\", \"__TIME__\"=>1391832857, \"__TAG__\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832858, {"count"=>2, "inspect"=>"[{\"key3\"=>\"val3\", \"__TIME__\"=>1391832858, \"__TAG__\"=>\"test.default\"}, {\"key3_\"=>\"val3_\", \"__TIME__\"=>1391832858, \"__TAG__\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832859, {"count"=>2, "inspect"=>"[{\"key4\"=>\"val4\", \"__TIME__\"=>1391832859, \"__TAG__\"=>\"test.default\"}, {\"key4_\"=>\"val4_\", \"__TIME__\"=>1391832859, \"__TAG__\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832860, {"count"=>2, "inspect"=>"[{\"key5\"=>\"val5\", \"__TIME__\"=>1391832860, \"__TAG__\"=>\"test.default\"}, {\"key5_\"=>\"val5_\", \"__TIME__\"=>1391832860, \"__TAG__\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832861, {"count"=>2, "inspect"=>"[{\"key6\"=>\"val6\", \"__TIME__\"=>1391832861, \"__TAG__\"=>\"test.default\"}, {\"key6_\"=>\"val6_\", \"__TIME__\"=>1391832861, \"__TAG__\"=>\"test.default\"}]"}],
+           ["filtered.test.default", 1391832862, {"count"=>2, "inspect"=>"[{\"key7\"=>\"val7\", \"__TIME__\"=>1391832862, \"__TAG__\"=>\"test.default\"}, {\"key7_\"=>\"val7_\", \"__TIME__\"=>1391832862, \"__TAG__\"=>\"test.default\"}]"}]]
+        )
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,63 @@
+require 'fluent/test'
+require 'fluent/plugin/out_unit_time_filter'
+require 'tempfile'
+require 'time'
+
+RSpec.configure do |config|
+  config.before(:all) do
+    Fluent::Test.setup
+  end
+end
+
+def tempfile(content)
+  Tempfile.open("#{File.basename __FILE__}.#{$$}") do |f|
+    f << content
+    f.flush
+    f.rewind
+    yield(f)
+  end
+end
+
+def run_driver(options = {})
+  options = options.dup
+
+  filter = options[:filter] || (<<-EOS)
+proc {|rs|
+  {
+    'count'   => rs.count,
+    'inspect' => rs.inspect
+  }
+}
+  EOS
+
+  option_keys = [
+    :unit_sec,
+    :prefix,
+    :emit_each_tag,
+    :pass_hash_row,
+    :hash_row_time_key,
+    :hash_row_tag_key,
+  ]
+
+  additional_options = option_keys.map {|key|
+    if options[key]
+      "#{key} #{options[key]}"
+    end
+  }.join("\n")
+
+  tag = options[:tag] || 'test.default'
+
+  tempfile(filter) do |f|
+    fluentd_conf = <<-EOS
+type unit_time_filter
+filter_path #{f.path}
+#{additional_options}
+    EOS
+
+    driver = Fluent::Test::OutputTestDriver.new(Fluent::UnitTimeFilterOutput, tag).configure(fluentd_conf)
+
+    driver.run do
+      yield(driver)
+    end
+  end
+end

metadata

@@ -0,0 +1,113 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-unit-time-filter
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Genki Sugawara
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-02-09 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: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !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: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.11.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.11.0
+description: Plug-in to aggregate by unit time
+email:
+- sgwr_dts@yahoo.co.jp
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- fluent-plugin-unit-time-filter.gemspec
+- lib/fluent/plugin/out_unit_time_filter.rb
+- lib/fluent/plugin/unit_time_filter_buffer.rb
+- spec/out_unit_time_filter_spec.rb
+- spec/spec_helper.rb
+homepage: https://bitbucket.org/winebarrel/fluent-plugin-unit-time-filter
+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: 
+rubygems_version: 2.0.14
+signing_key: 
+specification_version: 4
+summary: Plug-in to aggregate by unit time
+test_files:
+- spec/out_unit_time_filter_spec.rb
+- spec/spec_helper.rb