fluent-plugin-retag 0.0.1

data/.gitignore

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

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,13 @@
+Copyright (c) 2013- Masahiro Yamauchi
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

data/README.md

@@ -0,0 +1,54 @@
+# fluent-plugin-retag
+
+## Overview
+
+Output plugin only retagging.
+
+* simple retag
+* remove tag prefix
+* add tag prefix
+
+## Configuration
+
+### Simple Configuration
+
+To retag foo.bar to hoge.fuga:
+
+    <match foo.bar>
+      type retag
+      tag hoge.fuga
+    </match>
+
+To retag foo.bar.** to xyz.bar.**:
+
+    <match foo.bar>
+      type retag
+      remove_prefix foo
+      add_prefix xyz
+    </match>
+
+### Useful Configuration
+
+If you want to use branch condition, it is useful to use out_copy with fluent-plugin-retag.
+
+    <match foo.bar.**>
+      type copy
+      <store>
+        type retag
+        add_prefix copied
+      </store>
+      <store>
+        ...
+      </store>
+    </match>
+
+    <match copied.foo.bar.**>
+      ...
+    </match>
+
+## Copyright
+
+* Copyright (c) 2013- Masahiro Yamauchi
+* License
+  * Apache License, Version 2.0
+

data/Rakefile

@@ -0,0 +1,10 @@
+require "bundler/gem_tasks"
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+task :default => :test
+

data/fluent-plugin-retag.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+Gem::Specification.new do |gem|
+  gem.name          = "fluent-plugin-retag"
+  gem.version       = "0.0.1"
+  gem.authors       = ["Masahiro Yamauchi"]
+  gem.email         = ["sgt.yamauchi@gmail.com"]
+  gem.description   = %q{Output filter plugin to retag}
+  gem.summary       = %q{Output filter plugin to retag}
+  gem.homepage      = "https://github.com/algas/fluent-plugin-retag"
+
+  gem.rubyforge_project = "fluent-plugin-retag"
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+  gem.add_development_dependency "fluentd"
+  gem.add_runtime_dependency "fluentd"
+end

data/lib/fluent/plugin/out_retag.rb

@@ -0,0 +1,48 @@
+class Fluent::RetagOutput < Fluent::Output
+  Fluent::Plugin.register_output('retag', self)
+
+  config_param :tag, :string, :default => nil
+  config_param :remove_prefix, :string, :default => nil
+  config_param :add_prefix, :string, :default => nil
+
+  def configure(conf)
+    super
+    if not @tag and not @remove_prefix and not @add_prefix
+      raise Fluent::ConfigError, "missing both of remove_prefix and add_prefix"
+    end
+    if @tag and (@remove_prefix or @add_prefix)
+      raise Fluent::ConfigError, "both of tag and remove_prefix/add_prefix must not be specified"
+    end
+    if @remove_prefix
+      @removed_prefix_string = @remove_prefix + '.'
+      @removed_length = @removed_prefix_string.length
+    end
+    if @add_prefix
+      @added_prefix_string = @add_prefix + '.'
+    end
+  end
+
+  def emit(tag, es, chain)
+    tag = if @tag
+            @tag
+          else
+            if @remove_prefix and
+                ( (tag.start_with?(@removed_prefix_string) and tag.length > @removed_length) or tag == @remove_prefix)
+              tag = tag[@removed_length..-1]
+            end 
+            if @add_prefix 
+              tag = if tag and tag.length > 0
+                      @added_prefix_string + tag
+                    else
+                      @add_prefix
+                    end
+            end
+            tag
+          end
+    es.each do |time,record|
+      Fluent::Engine.emit(tag, time, record)
+    end
+    chain.next
+  end
+end
+

data/test/helper.rb

@@ -0,0 +1,29 @@
+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'
+unless ENV.has_key?('VERBOSE')
+  nulllogger = Object.new
+  nulllogger.instance_eval {|obj|
+    def method_missing(method, *args)
+      # pass
+    end
+  }
+  $log = nulllogger
+end
+
+require 'fluent/plugin/out_retag'
+
+class Test::Unit::TestCase
+end
+

data/test/plugin/test_out_retag.rb

@@ -0,0 +1,96 @@
+require 'helper'
+
+class RetagOutputTest < Test::Unit::TestCase
+  def setup
+    Fluent::Test.setup
+  end
+
+  CONFIG = %[
+    tag hoge
+  ]
+  CONFIG2 = %[
+    remove_prefix foo
+  ]
+  CONFIG3 = %[
+    remove_prefix foo
+    add_prefix head
+  ]
+
+  def create_driver(conf = CONFIG, tag='test')
+    Fluent::Test::OutputTestDriver.new(Fluent::RetagOutput, tag).configure(conf)
+  end
+
+  def test_configure
+    assert_raise(Fluent::ConfigError) {
+      create_driver ('')
+    }
+    assert_raise(Fluent::ConfigError) {
+      create_driver %[
+        tag a
+        remove_prefix b
+      ]
+    }
+    assert_raise(Fluent::ConfigError) {
+      create_driver %[
+        tag a
+        add_prefix c
+      ]
+    }
+    d1 = create_driver %[
+      tag a
+    ]
+    d1.instance.inspect
+    assert_equal 'a', d1.instance.tag
+    d2 = create_driver %[
+      remove_prefix b
+      add_prefix c
+    ]
+    d2.instance.inspect
+    assert_equal 'b', d2.instance.remove_prefix
+    assert_equal 'c', d2.instance.add_prefix
+  end
+
+  def test_emit
+    d = create_driver(CONFIG, 'foo.bar')
+    time = Time.parse('2011-03-11 14:46:01').to_i
+    d.run do
+      d.emit({'a' => 'b'})
+      d.emit({'c' => 'd'})
+    end
+    emits = d.emits
+    assert_equal 2, emits.length
+    p emits[0]
+    assert_equal 'hoge', emits[0][0]
+    assert_equal 'b', emits[0][2]['a']
+  end
+
+  def test_emit2
+    d = create_driver(CONFIG2, 'foo.bar')
+    time = Time.parse('2011-03-11 14:46:01').to_i
+    d.run do
+      d.emit({'a' => 'b'})
+      d.emit({'c' => 'd'})
+    end
+    emits = d.emits
+    assert_equal 2, emits.length
+    p emits[0]
+    assert_equal 'bar', emits[0][0]
+    assert_equal 'b', emits[0][2]['a']
+  end
+
+  def test_emit3
+    d = create_driver(CONFIG3, 'foo.bar')
+    time = Time.parse('2011-03-11 14:46:01').to_i
+    d.run do
+      d.emit({'a' => 'b'})
+      d.emit({'c' => 'd'})
+    end
+    emits = d.emits
+    assert_equal 2, emits.length
+    p emits[0]
+    assert_equal 'head.bar', emits[0][0]
+    assert_equal 'b', emits[0][2]['a']
+  end
+
+end
+

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-retag
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Masahiro Yamauchi
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-02-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Output filter plugin to retag
+email:
+- sgt.yamauchi@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- fluent-plugin-retag.gemspec
+- lib/fluent/plugin/out_retag.rb
+- test/helper.rb
+- test/plugin/test_out_retag.rb
+homepage: https://github.com/algas/fluent-plugin-retag
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: fluent-plugin-retag
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Output filter plugin to retag
+test_files:
+- test/helper.rb
+- test/plugin/test_out_retag.rb