add comment option

README.md

@@ -23,6 +23,10 @@ This plugin does not emit, just writes into the log file as
 
  One of num/byte. Default: num
 
+- comment
+
+ Just a comment. You may use this to mark where you put flowcounter_simple.
+
 ## Relatives
 
 There is a [fluent-plugin-flowcounter](https://github.com/tagomoris/fluent-plugin-flowcounter), 

lib/fluent/plugin/out_flowcounter_simple.rb

@@ -3,6 +3,7 @@ class Fluent::FlowCounterSimpleOutput < Fluent::Output
 
  config_param :indicator, :string, :default => 'num'
  config_param :unit, :string, :default => 'second'
+ config_param :comment, :string, :default => nil
 
  attr_accessor :last_checked
 
@@ -35,6 +36,13 @@ def configure(conf)
  raise RuntimeError, "@unit must be one of second/minute/hour/day"
  end
 
+ @output_proc =
+ if @comment
+ Proc.new {|count| "plugin:out_flowcounter_simple\tcount:#{count}\tindicator:#{@indicator}\tunit:#{@unit}\tcomment:#{@comment}" }
+ else
+ Proc.new {|count| "plugin:out_flowcounter_simple\tcount:#{count}\tindicator:#{@indicator}\tunit:#{@unit}" }
+ end
+
  @count = 0
  @mutex = Mutex.new
  end
@@ -59,7 +67,7 @@ def countup(count)
  def flush_emit(step)
  count, @count = @count, 0
  if count > 0
- $log.info "plugin:out_flowcounter_simple\tcount:#{count}\tindicator:#{@indicator}\tunit:#{@unit}"
+ $log.info @output_proc.call(count)
  end
  end
 

test/plugin/test_out_flowcounter_simple.rb

@@ -54,6 +54,19 @@ def test_byte
  assert( log.include?("count:3360"), log )
  end
 
+ def test_comment
+ d1 = create_driver(CONFIG + %[comment foobar], 'test.tag1')
+ d1.run do
+ 1.times do
+ d1.emit({'message'=> 'a' * 100})
+ d1.emit({'message'=> 'b' * 100})
+ d1.emit({'message'=> 'c' * 100})
+ end
+ end
+ log = capture_log { d1.instance.flush_emit(60) }
+ assert( log.include?("comment:foobar"), log )
+ end
+
 private
 
  def capture_log