fluent-plugin-random 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 27ff303f3039a2c269455b9654b316cd30fc6c2b
+  data.tar.gz: bdb258ea93944217628e46df2f591f424811c634
+SHA512:
+  metadata.gz: 8c09b999b03c61be7a1ae1e5794dd172ec965d59ae988ac67ae77d110d2ee31f6b369bbdbc40dd307aeca727f671022e4c070ce7717fbe5030486f964f4db893
+  data.tar.gz: 707f0241898c6bf9b78bf376bc9b2b15f1052df1a54fc49c8b7d2ab6b191abbcf0e7f81e690f18d1baec9dd17cebc18735a81e1379e5f878f9a7d6239b8c91ed

data/Gemfile

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

data/README.md

@@ -0,0 +1,47 @@
+fluent-plugin-random
+====================
+
+(Yet another Fluentd plugin)
+
+What
+----
+
+Allows to replace generate random value (only in integer at the time of this README) in a the specified key.
+
+How
+---
+
+You want to define :
+- Output field : **field**.
+- Integer interval : **integer** (syntax is the Ruby range..syntax)
+
+More random type could come in the future (float, string)
+
+Examples
+========
+
+Example 0
+---------
+
+```
+<match *.test>
+    type random
+    add_tag_prefix random.
+    integer 15..100
+    field key1
+</match>
+```
+
+Example of records :
+```
+{
+    'foo' => "bar"
+}
+```
+... *can* output :
+```
+{
+    'key1' => 42,
+    'foo' => "bar"
+}
+```

data/Rakefile

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

data/fluent-plugin-random.gemspec

@@ -0,0 +1,19 @@
+Gem::Specification.new do |spec|
+  spec.name          = "fluent-plugin-random"
+  spec.version       = "0.0.1"
+  spec.authors       = ["Neozaru"]
+  spec.email         = ["neozaru@mailoo.org"]
+  spec.description   = %q{Fluentd custom plugin to generate random values}
+  spec.summary       = %q{Fluentd custom plugin to generate random values}
+  spec.homepage      = "https://github.com/Neozaru/fluent-plugin-random.git"
+  spec.license       = "WTFPL"
+
+  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_development_dependency "bundler"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "fluentd"
+end

data/lib/fluent/plugin/out_random.rb

@@ -0,0 +1,73 @@
+# coding: utf-8
+require "securerandom"
+
+module Fluent
+  class RandomOutput < Output
+    include Fluent::HandleTagNameMixin
+
+    Fluent::Plugin.register_output('random', self)
+
+    config_param :integer, :string, :default => nil
+    config_param :field, :string, :default => nil
+
+
+    def parse_integer_field(interval)
+      lower_upper = interval.split("..", 2).map{|value| Integer(value) }
+      if (lower_upper.length != 2 || !lower_upper[0].is_a?(Integer) || !lower_upper[1].is_a?(Integer))
+        raise "random: Bad value for 'integer' (value: '#{interval}')"
+      end
+      return lower_upper
+    end
+
+    def check_valid_interval(lower, upper)
+      if (lower >= upper)
+        raise "random: Bad interval for 'integer' (#{lower} >= #{upper})"
+      end
+    end
+
+
+    def configure(conf)
+      super
+
+      if (field.nil? || integer.nil?)
+        raise ConfigError, "random: 'field' is required to be set."
+      end
+
+      if (integer.nil?)
+        raise ConfigError, "random: 'integer' is required to be set."
+      end
+
+      @integer_interval = parse_integer_field(integer)
+      check_valid_interval(@integer_interval[0], @integer_interval[1])
+      @field = field
+      @random  = Random.new()
+    end
+
+    def emit(tag, es, chain)
+      es.each { |time, record|
+        t = tag.dup
+        filter_record(t, time, record)
+        Engine.emit(t, time, record)
+      }
+
+      chain.next
+    end
+
+    private
+
+    def filter_record(tag, time, record)
+      super(tag, time, record)
+      record[@field] = process()
+    end
+
+    def process()
+      
+      return generate_integer(@integer_interval[0], @integer_interval[1])
+    end
+
+    def generate_integer(lower, upper)
+      return @random.rand(lower..upper)
+    end
+
+  end
+end

data/test/plugin/test_out_random.rb

@@ -0,0 +1,120 @@
+# coding: utf-8
+
+require 'test_helper'
+require 'fluent/plugin/out_random'
+
+class RandomOutputTest < Test::Unit::TestCase
+  def setup
+    Fluent::Test.setup
+    @p = create_driver(%[
+      strict true
+      add_tag_prefix random.
+      field key1
+      integer 1..10
+    ])
+  end
+
+  def create_driver(conf, tag = 'test')
+    Fluent::Test::OutputTestDriver.new(
+      Fluent::RandomOutput, tag
+    ).configure(conf)
+  end
+
+  def test_unit_parse_integer
+    assert_raise do @p.instance.parse_integer_field("") end
+    assert_raise do @p.instance.parse_integer_field("1.10") end
+    assert_raise do @p.instance.parse_integer_field("1..10,1") end
+    assert_raise do @p.instance.parse_integer_field("10") end
+    assert_raise do @p.instance.parse_integer_field("1..10..11") end
+    assert_raise do @p.instance.parse_integer_field("1..") end
+    assert_raise do @p.instance.parse_integer_field("..10") end
+    assert_raise do @p.instance.parse_integer_field("foo..10") end
+    assert_raise do @p.instance.parse_integer_field("1..bar") end
+    assert_raise do @p.instance.parse_integer_field("foo") end
+    assert_raise do @p.instance.parse_integer_field("foo..bar") end
+
+    lower_upper = @p.instance.parse_integer_field("1..10")
+    assert_equal 1, lower_upper[0]
+    assert_equal 10, lower_upper[1]
+
+    lower_upper = @p.instance.parse_integer_field("10..1")
+    assert_equal 10, lower_upper[0]
+    assert_equal 1, lower_upper[1]
+
+    lower_upper = @p.instance.parse_integer_field("-10000000000..10000000000")
+    assert_equal -10000000000, lower_upper[0]
+    assert_equal 10000000000, lower_upper[1]
+
+    lower_upper = @p.instance.parse_integer_field("-1000000000000000000000000000000..1000000000000000000000000000000")
+    assert_equal -1000000000000000000000000000000, lower_upper[0]
+    assert_equal 1000000000000000000000000000000, lower_upper[1]
+  end
+
+  def test_unit_check_valid_interval()
+    assert_raise do @p.instance.check_valid_interval(10,1) end
+    assert_raise do @p.instance.check_valid_interval(10,-1) end
+
+    @p.instance.check_valid_interval(-50,10)
+    @p.instance.check_valid_interval(1,10)
+    @p.instance.check_valid_interval(-1000000000000000000000000000000, 1000000000000000000000000000000)
+  end
+
+  def test_configure_on_success
+    # All set
+    d = create_driver(%[
+      strict true
+      add_tag_prefix random.
+      field key1
+      integer 1..10
+    ])
+
+    assert_equal 'key1',    d.instance.field
+  end
+
+  def test_configure_on_failure
+    # when mandatory keys not set
+    assert_raise(Fluent::ConfigError) do
+      create_driver(%[
+        blah blah
+      ])
+    end
+
+    # 'field' is missing
+    assert_raise(Fluent::ConfigError) do
+      create_driver(%[
+        add_tag_prefix random.
+        integer 1..10
+      ])
+    end
+
+    # 'integer' is missing
+    assert_raise(Fluent::ConfigError) do
+      create_driver(%[
+        add_tag_prefix random.
+        field key1
+      ])
+    end
+  end
+
+  def test_emit
+    d = create_driver(%[
+      add_tag_prefix random.
+      field key1
+      integer 1..10
+    ])
+
+    record = {
+      'foo' => 'bar'
+    }
+
+    d.run { d.emit(record) }
+    emits = d.emits
+
+    assert_equal 1,           emits.count
+    assert_equal 'random.test', emits[0][0]
+    assert_equal 'bar', emits[0][2]['foo']
+    assert_includes 1..10, emits[0][2]['key1']
+
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,19 @@
+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
+
+class Test::Unit::TestCase
+end

metadata

@@ -0,0 +1,95 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-random
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Neozaru
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-04-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  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: 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: fluentd
+  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 custom plugin to generate random values
+email:
+- neozaru@mailoo.org
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- README.md
+- Rakefile
+- fluent-plugin-random.gemspec
+- lib/fluent/plugin/out_random.rb
+- test/plugin/test_out_random.rb
+- test/test_helper.rb
+homepage: https://github.com/Neozaru/fluent-plugin-random.git
+licenses:
+- WTFPL
+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.4.5
+signing_key: 
+specification_version: 4
+summary: Fluentd custom plugin to generate random values
+test_files:
+- test/plugin/test_out_random.rb
+- test/test_helper.rb