fluent-plugin-stats 0.3.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 5bc9f72a5a727bc1397f45b015193d8762b10533
+  data.tar.gz: a2ba051b18070f6cf5518416c50330495fa56829
+SHA512:
+  metadata.gz: 212b0f23da7de8e80edfa0a01de810730d6f2e23adc83b694a63214197f04b9ae58a0b1091754e0745cf396567a4e60b3f9ef1e0ebfd464be013816bd80d69dc
+  data.tar.gz: 577089bb6d526f10d4911891e0fb8284073cc97d6a91a9439dfaedf66af3e4ea41a34c772508df120f5e03ccf4e8aa8ca837b195ef89ea391ec07e3fbc53b899

data/.coveralls.yml

@@ -0,0 +1 @@
+repo_token: i4DJCtdksuIwhBck1tukIjzKoMCxWIIvQ

data/.gitignore

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

data/.rdebugrc

@@ -0,0 +1,4 @@
+set autolist
+set autoeval
+set autoreload
+

data/.rspec

@@ -0,0 +1,2 @@
+--colour
+--format documentation

data/.travis.yml

@@ -0,0 +1,6 @@
+rvm:
+  - 1.9.2
+  - 1.9.3
+  - 2.0.0
+gemfile:
+  - Gemfile

data/CHANGELOG.md

@@ -0,0 +1,52 @@
+## 0.3.0 (2013/12/27)
+
+Changes
+
+- Rename fluent-plugin-stats from fluent-plugin-calc
+
+## 0.2.0 (2013/12/12)
+
+Enhancement:
+
+- Add `remove_tag_prefix` option
+
+## 0.1.2 (2013/10/10)
+
+Enhancement:
+
+- Add `zero_emit` option to emit 0 on the next interval like datacounter plugin
+
+## 0.1.1 (2013/09/02)
+
+Fixes
+
+- Fix the case when not record is found.
+
+## 0.1.0 (2013/09/02)
+
+Changes
+
+- Accept string data by `to_f`.
+
+## 0.0.5 (2013/09/02)
+
+Enhancement:
+
+- add `sum_keys`, `max_keys`, `min_keys`, `avg_keys`.
+
+## 0.0.4 (2013/08/30)
+
+Enhancement:
+
+- add `sum_suffix`, `max_suffix`, `min_suffix`, `avg_suffix`.
+
+## 0.0.2 (2013/05/06)
+
+Bugfixes:
+
+- Fix so that @avg can be an option
+
+## 0.0.1 (2013/05/05)
+
+First version
+

data/Gemfile

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

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Naotoshi SEO
+
+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,126 @@
+# fluent-plugin-stats [![Build Status](https://secure.travis-ci.org/sonots/fluent-plugin-stats.png?branch=master)](http://travis-ci.org/sonots/fluent-plugin-stats)
+
+Fluentd plugin to calculate statistics such as sum, max, min, avg.
+
+## Configuration
+
+### Example 1
+
+Get sum for xxx\_count, max for xxx\_max, min for xxx\_min, avg for xxx\_avg
+
+    <match foo.**>
+      type stats
+      interval 5s
+      add_tag_prefix stats
+
+      sum .*_count$
+      max .*_max$
+      min .*_min$
+      avg .*_avg$
+    </match>
+
+Assuming following inputs are coming:
+
+    foo.bar: {"4xx_count":1,"5xx_count":2","reqtime_max":12083,"reqtime_min":10,"reqtime_avg":240.46}
+    foo.bar: {"4xx_count":4,"5xx_count":2","reqtime_max":24831,"reqtime_min":82,"reqtime_avg":300.46}
+
+then output bocomes as belows: 
+
+    stats.foo.bar: {"4xx_count":5,"5xx_count":4","reqtime_max":24831,"reqtime_min":10,"reqtime_avg":270.46}
+
+### Example 2
+
+Get sum, max, min, avg for the same key
+
+    <match foo.**>
+      type stats
+      interval 5s
+      add_tag_prefix stats
+
+      sum ^reqtime$
+      max ^reqtime$
+      min ^reqtime$
+      avg ^reqtime$
+      sum_suffix _sum
+      max_suffix _max
+      min_suffix _min
+      avg_suffix _avg
+    </match>
+
+Assuming following inputs are coming:
+
+    foo.bar: {"reqtime":1.000}
+    foo.bar: {"reqtime":2.000}
+
+then output bocomes as belows: 
+
+    stats.foo.bar: {"reqtime_sum":3.000,"reqtime_max":2.000,"reqtime_min":1.000,"reqtime_avg":1.500}
+
+## Parameters
+
+- sum, min, max, avg
+
+    Target of calculation. Specify input keys by a regular expression
+
+- sum\_keys, min\_keys, max\_keys, avg\_keys
+
+    Target of calculation. Specify input keys by a string separated by , (comma) such as
+
+        sum_keys 4xx_count,5xx_count
+
+- sum\_suffix, min\_suffix, max\_suffix, avg\_suffix
+
+    Add a suffix to keys of the output record
+
+- interval
+
+    The interval to calculate in seconds. Default is 5s. 
+
+- tag
+
+    The output tag name. Required for aggregate `all`. 
+
+- add_tag_prefix
+
+    Add tag prefix for output message. Default: 'stats'
+
+- remove_tag_prefix
+
+    Remove tag prefix for output message. 
+
+- aggragate
+    
+    Statsulate by each `tag` or `all`. The default value is `tag`.
+
+- store_file
+
+    Store internal data into a file of the given path on shutdown, and load on starting.
+
+- zero_emit
+
+    Emit 0 on the next interval. This is useful for some software which requires to reset data such as [GrowthForecast](http://kazeburo.github.io/GrowthForecast/) . 
+
+        stats.foo.bar: {"4xx_count":5,"5xx_count":4","reqtime_max":24831,"reqtime_min":10,"reqtime_avg":270.46}
+        # after @interval later
+        stats.foo.bar: {"4xx_count":0,"5xx_count":0","reqtime_max":0,"reqtime_min":0,"reqtime_avg":0}
+
+## ChangeLog
+
+See [CHANGELOG.md](CHANGELOG.md) for details.
+
+## ToDo
+
+Get the number of denominator to calculate `avg` from input json field. 
+
+## 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)
+
+## Copyright
+
+Copyright (c) 2013 Naotoshi Seo. See [LICENSE](LICENSE) for details.
+

data/Rakefile

@@ -0,0 +1,15 @@
+# encoding: utf-8
+require "bundler/gem_tasks"
+
+require 'rspec/core'
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec) do |spec|
+  spec.pattern = FileList['spec/**/*_spec.rb']
+end
+task :default => :spec
+
+desc 'Open an irb session preloaded with the gem library'
+task :console do
+    sh 'irb -rubygems -I lib'
+end
+task :c => :console

data/fluent-plugin-stats.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "fluent-plugin-stats"
+  s.version     = "0.3.0"
+  s.authors     = ["Naotoshi Seo"]
+  s.email       = ["sonots@gmail.com"]
+  s.homepage    = "https://github.com/sonots/fluent-plugin-stats"
+  s.summary     = "Fluentd plugin to calculate statistics such as sum, max, min, avg"
+  s.description = s.summary
+  s.licenses    = ["MIT"]
+
+  s.rubyforge_project = "fluent-plugin-stats"
+
+  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 "rspec"
+  s.add_development_dependency "pry"
+  s.add_development_dependency "pry-nav"
+end

data/lib/fluent/plugin/out_stats.rb

@@ -0,0 +1,312 @@
+# encoding: UTF-8
+class Fluent::StatsOutput < Fluent::Output
+  Fluent::Plugin.register_output('stats', self)
+
+  def initialize
+    super
+    require 'pathname'
+  end
+
+  config_param :sum, :string, :default => nil
+  config_param :max, :string, :default => nil
+  config_param :min, :string, :default => nil
+  config_param :avg, :string, :default => nil
+  config_param :sum_keys, :string, :default => nil
+  config_param :max_keys, :string, :default => nil
+  config_param :min_keys, :string, :default => nil
+  config_param :avg_keys, :string, :default => nil
+  config_param :sum_suffix, :string, :default => ""
+  config_param :max_suffix, :string, :default => ""
+  config_param :min_suffix, :string, :default => ""
+  config_param :avg_suffix, :string, :default => ""
+  config_param :interval, :time, :default => 5
+  config_param :tag, :string, :default => nil
+  config_param :add_tag_prefix, :string, :default => nil
+  config_param :remove_tag_prefix, :string, :default => nil
+  config_param :aggregate, :string, :default => 'tag'
+  config_param :store_file, :string, :default => nil
+  config_param :zero_emit, :bool, :default => false
+
+  attr_accessor :matches
+  attr_accessor :saved_duration
+  attr_accessor :saved_at
+  attr_accessor :last_checked
+
+  def configure(conf)
+    super
+
+    @interval = @interval.to_i
+    @sum = Regexp.new(@sum) if @sum
+    @max = Regexp.new(@max) if @max
+    @min = Regexp.new(@min) if @min
+    @avg = Regexp.new(@avg) if @avg
+    @sum_keys = @sum_keys ? @sum_keys.split(',') : []
+    @max_keys = @max_keys ? @max_keys.split(',') : []
+    @min_keys = @min_keys ? @min_keys.split(',') : []
+    @avg_keys = @avg_keys ? @avg_keys.split(',') : []
+
+    unless ['tag', 'all'].include?(@aggregate)
+      raise Fluent::ConfigError, "aggregate allows tag/all"
+    end
+
+    case @aggregate
+    when 'all'
+      raise Fluent::ConfigError, "tag must be specified for aggregate all" if @tag.nil?
+    end
+
+    if @tag.nil? and @add_tag_prefix.nil? and @remove_tag_prefix.nil?
+      @add_tag_prefix = 'stats' # not ConfigError for lower version compatibility
+    end
+
+    @tag_prefix = "#{@add_tag_prefix}." if @add_tag_prefix
+    @tag_prefix_match = "#{@remove_tag_prefix}." if @remove_tag_prefix
+    @tag_proc =
+      if @tag
+        Proc.new {|tag| @tag }
+      elsif @tag_prefix and @tag_prefix_match
+        Proc.new {|tag| "#{@tag_prefix}#{lstrip(tag, @tag_prefix_match)}" }
+      elsif @tag_prefix_match
+        Proc.new {|tag| lstrip(tag, @tag_prefix_match) }
+      elsif @tag_prefix
+        Proc.new {|tag| "#{@tag_prefix}#{tag}" }
+      else
+        Proc.new {|tag| tag }
+      end
+
+    @matches = {}
+    @mutex = Mutex.new
+  end
+
+  def initial_matches(prev_matches = nil)
+    if @zero_emit && prev_matches
+      matches = {}
+      prev_matches.keys.each do |tag|
+        next unless prev_matches[tag][:count] > 0 # Prohibit to emit anymore
+        matches[tag] = { :count => 0, :sum => {}, :max => {}, :min => {}, :avg => {} }
+        # ToDo: would want default configuration for :max, :min
+        prev_matches[tag][:sum].keys.each {|key| matches[tag][:sum][key] = 0 }
+        prev_matches[tag][:max].keys.each {|key| matches[tag][:max][key] = 0 }
+        prev_matches[tag][:min].keys.each {|key| matches[tag][:min][key] = 0 }
+        prev_matches[tag][:avg].keys.each {|key| matches[tag][:avg][key] = 0 }
+      end
+      matches
+    else
+      {}
+    end
+  end
+
+  def start
+    super
+    load_status(@store_file, @interval) if @store_file
+    @watcher = Thread.new(&method(:watcher))
+  end
+
+  def shutdown
+    super
+    @watcher.terminate
+    @watcher.join
+    save_status(@store_file) if @store_file
+  end
+
+  # Called when new line comes. This method actually does not emit
+  def emit(tag, es, chain)
+    tag = 'all' if @aggregate == 'all'
+    # stats
+    matches = { :count => 0, :sum => {}, :max => {}, :min => {}, :avg => {} }
+    es.each do |time, record|
+      @sum_keys.each do |key|
+        next unless record[key] and value = record[key].to_f
+        matches[:sum][key] = sum(matches[:sum][key], value)
+      end
+      @max_keys.each do |key|
+        next unless record[key] and value = record[key].to_f
+        matches[:max][key] = max(matches[:max][key], value)
+      end
+      @min_keys.each do |key|
+        next unless record[key] and value = record[key].to_f
+        matches[:min][key] = min(matches[:min][key], value)
+      end
+      @avg_keys.each do |key|
+        next unless record[key] and value = record[key].to_f
+        matches[:avg][key] = sum(matches[:avg][key], value)
+      end
+      record.keys.each do |key|
+        value = record[key].to_f
+        if @sum and @sum.match(key)
+          matches[:sum][key] = sum(matches[:sum][key], value)
+        end
+        if @max and @max.match(key)
+          matches[:max][key] = max(matches[:max][key], value)
+        end
+        if @min and @min.match(key)
+          matches[:min][key] = min(matches[:min][key], value)
+        end
+        if @avg and @avg.match(key)
+          matches[:avg][key] = sum(matches[:avg][key], value) # sum yet
+        end
+      end if @sum || @max || @min || @avg
+      matches[:count] += 1
+    end
+
+    # thread safe merge
+    @matches[tag] ||= { :count => 0, :sum => {}, :max => {}, :min => {}, :avg => {} }
+    @mutex.synchronize do
+      matches[:sum].keys.each do |key|
+        @matches[tag][:sum][key] = sum(@matches[tag][:sum][key], matches[:sum][key])
+      end
+      matches[:max].keys.each do |key|
+        @matches[tag][:max][key] = max(@matches[tag][:max][key], matches[:max][key])
+      end
+      matches[:min].keys.each do |key|
+        @matches[tag][:min][key] = min(@matches[tag][:min][key], matches[:min][key])
+      end
+      matches[:avg].keys.each do |key|
+        @matches[tag][:avg][key] = sum(@matches[tag][:avg][key], matches[:avg][key]) # sum yet
+      end
+      @matches[tag][:count] += matches[:count]
+    end
+
+    chain.next
+  rescue => e
+    $log.warn "#{e.class} #{e.message} #{e.backtrace.first}"
+  end
+
+  # thread callback
+  def watcher
+    # instance variable, and public accessable, for test
+    @last_checked ||= Fluent::Engine.now
+    while true
+      sleep 0.5
+      begin
+        if Fluent::Engine.now - @last_checked >= @interval
+          now = Fluent::Engine.now
+          flush_emit(now - @last_checked)
+          @last_checked = now
+        end
+      rescue => e
+        $log.warn "#{e.class} #{e.message} #{e.backtrace.first}"
+      end
+    end
+  end
+
+  # This method is the real one to emit
+  def flush_emit(step)
+    time = Fluent::Engine.now
+    flushed_matches, @matches = @matches, initial_matches(@matches)
+
+    flushed_matches.keys.each do |tag|
+      matches = flushed_matches[tag]
+      output = generate_output(matches)
+      emit_tag = @tag_proc.call(tag)
+      Fluent::Engine.emit(emit_tag, time, output) if output
+    end
+  end
+
+  def generate_output(matches)
+    return nil if matches.empty?
+    output = {}
+    matches[:sum].keys.each do |key|
+      output[key + @sum_suffix] = matches[:sum][key]
+    end
+    matches[:max].keys.each do |key|
+      output[key + @max_suffix] = matches[:max][key]
+    end
+    matches[:min].keys.each do |key|
+      output[key + @min_suffix] = matches[:min][key]
+    end
+    matches[:avg].keys.each do |key|
+      output[key + @avg_suffix] = matches[:avg][key]
+      output[key + @avg_suffix] /= matches[:count].to_f if matches[:count] > 0
+    end
+    output
+  end
+
+  def sum(a, b)
+    [a, b].compact.inject(:+)
+  end
+
+  def max(a, b)
+    [a, b].compact.max
+  end
+
+  def min(a, b)
+    [a, b].compact.min
+  end
+
+  # Store internal status into a file
+  #
+  # @param [String] file_path
+  def save_status(file_path)
+    return unless file_path
+
+    begin
+      Pathname.new(file_path).open('wb') do |f|
+        @saved_at = Fluent::Engine.now
+        @saved_duration = @saved_at - @last_checked
+        Marshal.dump({
+          :matches          => @matches,
+          :saved_at         => @saved_at,
+          :saved_duration   => @saved_duration,
+          :aggregate        => @aggregate,
+          :sum              => @sum,
+          :max              => @max,
+          :min              => @min,
+          :avg              => @avg,
+        }, f)
+      end
+    rescue => e
+      $log.warn "out_stats: Can't write store_file #{e.class} #{e.message}"
+    end
+  end
+
+  # Load internal status from a file
+  #
+  # @param [String] file_path
+  # @param [Interger] interval
+  def load_status(file_path, interval)
+    return unless (f = Pathname.new(file_path)).exist?
+
+    begin
+      f.open('rb') do |f|
+        stored = Marshal.load(f)
+        if stored[:aggregate] == @aggregate and
+          stored[:sum] == @sum and
+          stored[:max] == @max and
+          stored[:min] == @min and
+          stored[:avg] == @avg
+
+          if !stored[:matches].empty? and !stored[:matches].first[1].has_key?(:max)
+            $log.warn "out_stats: stored data does not have compatibility with the current version. ignore stored data"
+            return
+          end
+
+          if Fluent::Engine.now <= stored[:saved_at] + interval
+            @matches = stored[:matches]
+            @saved_at = stored[:saved_at]
+            @saved_duration = stored[:saved_duration]
+            # for lower compatibility
+            if counts = stored[:counts]
+              @matches.keys.each {|tag| @matches[tag][:count] = counts[tag] }
+            end
+
+            # skip the saved duration to continue counting
+            @last_checked = Fluent::Engine.now - @saved_duration
+          else
+            $log.warn "out_stats: stored data is outdated. ignore stored data"
+          end
+        else
+          $log.warn "out_stats: configuration param was changed. ignore stored data"
+        end
+      end
+    rescue => e
+      $log.warn "out_stats: Can't load store_file #{e.class} #{e.message}"
+    end
+  end
+
+  private
+
+  def lstrip(string, substring)
+    string.index(substring) == 0 ? string[substring.size..-1] : string
+  end
+
+end

data/spec/out_stats_spec.rb

@@ -0,0 +1,307 @@
+# encoding: UTF-8
+require_relative 'spec_helper'
+
+class Fluent::Test::OutputTestDriver
+  def emit_with_tag(record, time=Time.now, tag = nil)
+    @tag = tag if tag
+    emit(record, time)
+  end
+end
+
+describe Fluent::StatsOutput do
+  before { Fluent::Test.setup }
+  CONFIG = %[]
+  let(:tag) { 'foo.bar' }
+  let(:driver) { Fluent::Test::OutputTestDriver.new(Fluent::StatsOutput, tag).configure(config) }
+
+  describe 'test configure' do
+    describe 'bad configuration' do
+      context 'invalid aggregate' do
+        let(:config) do
+          CONFIG + %[
+            aggregate foo
+          ]
+        end
+        it { expect { driver }.to raise_error(Fluent::ConfigError) }
+      end
+
+      context 'no tag for aggregate all' do
+        let(:config) do
+          CONFIG + %[
+            aggregate all
+          ]
+        end
+        it { expect { driver }.to raise_error(Fluent::ConfigError) }
+      end
+    end
+
+    describe 'good configuration' do
+      context "nothing" do
+        let(:config) { '' }
+        it { expect { driver }.to_not raise_error }
+      end
+
+      context 'sum/max/min/avg' do
+        let(:config) do
+          CONFIG + %[
+            sum _count$
+            max _max$
+            min _min$
+            avg _avg$
+          ]
+        end
+        it { expect { driver }.to_not raise_error }
+      end
+
+      context "check default" do
+        subject { driver.instance }
+        let(:config) { CONFIG }
+        its(:interval) { should == 5 }
+        its(:tag) { should be_nil }
+        its(:add_tag_prefix) { should == 'stats' }
+        its(:aggregate) { should == 'tag' }
+      end
+    end
+  end
+
+  describe 'test emit' do
+    let(:time) { Time.now.to_i }
+    let(:messages) do
+      [
+        {"4xx_count"=>"1","5xx_count"=>2,"reqtime_max"=>6,"reqtime_min"=>1,"reqtime_avg"=>3},
+        {"4xx_count"=>"2","5xx_count"=>2,"reqtime_max"=>5,"reqtime_min"=>2,"reqtime_avg"=>2},
+        {"4xx_count"=>"3","5xx_count"=>2,"reqtime_max"=>1,"reqtime_min"=>3,"reqtime_avg"=>4},
+      ]
+    end
+    let(:emit) do
+      driver.run { messages.each {|message| driver.emit(message, time) } }
+      driver.instance.flush_emit(0)
+    end
+    let(:empty_emit) do
+      driver.instance.flush_emit(0)
+    end
+
+    context 'sum/max/min/avg' do
+      let(:config) do
+        CONFIG + %[
+          sum _count$
+          max _max$
+          min _min$
+          avg _avg$
+        ]
+      end
+      before do
+        Fluent::Engine.stub(:now).and_return(time)
+        Fluent::Engine.should_receive(:emit).with("stats.#{tag}", time, {
+          "4xx_count"=>6,"5xx_count"=>6,"reqtime_max"=>6,"reqtime_min"=>1,"reqtime_avg"=>3.0
+        })
+      end
+      it { emit }
+    end
+
+    context 'zero_emit' do
+      let(:config) do
+        CONFIG + %[
+          sum _count$
+          max _max$
+          min _min$
+          avg _avg$
+          zero_emit true
+        ]
+      end
+      before do
+        Fluent::Engine.stub(:now).and_return(time)
+        Fluent::Engine.should_receive(:emit).with("stats.#{tag}", time, {
+          "4xx_count"=>6,"5xx_count"=>6,"reqtime_max"=>6,"reqtime_min"=>1,"reqtime_avg"=>3.0
+        })
+        Fluent::Engine.should_receive(:emit).with("stats.#{tag}", time, {
+          "4xx_count"=>0,"5xx_count"=>0,"reqtime_max"=>0,"reqtime_min"=>0,"reqtime_avg"=>0.0
+        })
+      end
+      it { emit; empty_emit }
+    end
+
+    context 'sum/max/min/avg_keys' do
+      let(:config) do
+        CONFIG + %[
+          sum_keys 4xx_count,5xx_count
+          max_keys reqtime_max
+          min_keys reqtime_min
+          avg_keys reqtime_avg,not_found
+        ]
+      end
+      before do
+        Fluent::Engine.stub(:now).and_return(time)
+        Fluent::Engine.should_receive(:emit).with("stats.#{tag}", time, {
+          "4xx_count"=>6,"5xx_count"=>6,"reqtime_max"=>6,"reqtime_min"=>1,"reqtime_avg"=>3.0
+        })
+      end
+      it { emit }
+    end
+
+    context 'sum/max/min/avg_suffix' do
+      let(:config) do
+        CONFIG + %[
+          sum ^(reqtime|reqsize)$
+          max ^reqtime$
+          min ^reqtime$
+          avg ^reqtime$
+          sum_suffix _sum
+          max_suffix _max
+          min_suffix _min
+          avg_suffix _avg
+        ]
+      end
+      let(:messages) do
+        [
+          {"reqtime"=>1.000,"reqsize"=>10},
+          {"reqtime"=>2.000,"reqsize"=>20},
+        ]
+    end
+      before do
+        Fluent::Engine.stub(:now).and_return(time)
+        Fluent::Engine.should_receive(:emit).with("stats.#{tag}", time, {
+          "reqtime_sum"=>3.000,"reqtime_max"=>2.000,"reqtime_min"=>1.000,"reqtime_avg"=>1.500,"reqsize_sum"=>30
+        })
+      end
+      it { emit }
+    end
+
+    context 'tag' do
+      let(:config) do
+        CONFIG + %[
+          tag foo
+          sum _count$
+        ]
+      end
+      before do
+        Fluent::Engine.stub(:now).and_return(time)
+        Fluent::Engine.should_receive(:emit).with("foo", time, {
+          "4xx_count"=>6,"5xx_count"=>6
+        })
+      end
+      it { emit }
+    end
+
+    context 'add_tag_prefix' do
+      let(:config) do
+        CONFIG + %[
+          add_tag_prefix foo
+          sum _count$
+        ]
+      end
+      before do
+        Fluent::Engine.stub(:now).and_return(time)
+        Fluent::Engine.should_receive(:emit).with("foo.#{tag}", time, {
+          "4xx_count"=>6,"5xx_count"=>6
+        })
+      end
+      it { emit }
+    end
+
+    context 'remove_tag_prefix' do
+      let(:config) do
+        CONFIG + %[
+          remove_tag_prefix foo
+          sum _count$
+        ]
+      end
+      before do
+        Fluent::Engine.stub(:now).and_return(time)
+        Fluent::Engine.should_receive(:emit).with("bar", time, {
+          "4xx_count"=>6,"5xx_count"=>6
+        })
+      end
+      it { emit }
+    end
+
+    context 'aggregate' do
+      let(:emit) do
+        driver.run { messages.each {|message| driver.emit_with_tag(message, time, 'foo.bar') } }
+        driver.run { messages.each {|message| driver.emit_with_tag(message, time, 'foo.bar2') } }
+        driver.instance.flush_emit(0)
+      end
+
+      context 'aggregate all' do
+        let(:config) do
+          CONFIG + %[
+          aggregate all
+          tag foo
+          sum _count$
+          max _max$
+          min _min$
+          avg _avg$
+        ]
+        end
+        before do
+          Fluent::Engine.stub(:now).and_return(time)
+          Fluent::Engine.should_receive(:emit).with("foo", time, {
+            "4xx_count"=>12,"5xx_count"=>12,"reqtime_max"=>6,"reqtime_min"=>1,"reqtime_avg"=>3.0
+          })
+        end
+        it { emit }
+      end
+
+      context 'aggregate tag' do
+        let(:config) do
+          CONFIG + %[
+          aggregate tag
+          add_tag_prefix stats
+          sum _count$
+          max _max$
+          min _min$
+          avg _avg$
+          ]
+        end
+        before do
+          Fluent::Engine.stub(:now).and_return(time)
+          Fluent::Engine.should_receive(:emit).with("stats.foo.bar", time, {
+            "4xx_count"=>6,"5xx_count"=>6,"reqtime_max"=>6,"reqtime_min"=>1,"reqtime_avg"=>3.0
+          })
+          Fluent::Engine.should_receive(:emit).with("stats.foo.bar2", time, {
+            "4xx_count"=>6,"5xx_count"=>6,"reqtime_max"=>6,"reqtime_min"=>1,"reqtime_avg"=>3.0
+          })
+        end
+        it { emit }
+      end
+    end
+
+    describe "store_file" do
+      let(:store_file) do
+        dirname = "tmp"
+        Dir.mkdir dirname unless Dir.exist? dirname
+        filename = "#{dirname}/test.dat"
+        File.unlink filename if File.exist? filename
+        filename
+      end
+
+      let(:config) do
+        CONFIG + %[
+          sum _count$
+          store_file #{store_file}
+        ]
+      end
+
+      it 'stored_data and loaded_data should equal' do
+        driver.run { messages.each {|message| driver.emit(message, time) } }
+        driver.instance.shutdown
+        stored_matches = driver.instance.matches
+        stored_saved_at = driver.instance.saved_at
+        stored_saved_duration = driver.instance.saved_duration
+        driver.instance.matches = {}
+        driver.instance.saved_at = nil
+        driver.instance.saved_duration = nil
+
+        driver.instance.start
+        loaded_matches = driver.instance.matches
+        loaded_saved_at = driver.instance.saved_at
+        loaded_saved_duration = driver.instance.saved_duration
+
+        loaded_matches.should == stored_matches
+        loaded_saved_at.should == stored_saved_at
+        loaded_saved_duration.should == stored_saved_duration
+      end
+    end
+  end
+end
+

data/spec/spec_helper.rb

@@ -0,0 +1,13 @@
+# encoding: UTF-8
+require 'rubygems'
+require 'bundler'
+Bundler.setup(:default, :test)
+Bundler.require(:default, :test)
+
+require 'fluent/test'
+require 'rspec'
+require 'pry'
+
+$TESTING=true
+$:.unshift File.join(File.dirname(__FILE__), '..', 'lib')
+require 'fluent/plugin/out_stats'

metadata

@@ -0,0 +1,130 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-stats
+version: !ruby/object:Gem::Version
+  version: 0.3.0
+platform: ruby
+authors:
+- Naotoshi Seo
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-12-27 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: rspec
+  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 plugin to calculate statistics such as sum, max, min, avg
+email:
+- sonots@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .coveralls.yml
+- .gitignore
+- .rdebugrc
+- .rspec
+- .travis.yml
+- CHANGELOG.md
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- fluent-plugin-stats.gemspec
+- lib/fluent/plugin/out_stats.rb
+- spec/out_stats_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/sonots/fluent-plugin-stats
+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-stats
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: Fluentd plugin to calculate statistics such as sum, max, min, avg
+test_files:
+- spec/out_stats_spec.rb
+- spec/spec_helper.rb