fluent-plugin-reassemble 0.0.1

checksums.yaml

@@ -0,0 +1,15 @@
+---
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    NmZmMjIwNzk2NjBlYzgwYzhkODgwYTc0ZDA5ZjE5NGI1NDkzZGFhOQ==
+  data.tar.gz: !binary |-
+    ZmIxMGNkNjljODVjNGUzY2RkNjZhM2RmYTNhZTcwN2ZhN2Q1OTE4Mw==
+!binary "U0hBNTEy":
+  metadata.gz: !binary |-
+    ZTNhNmVmYzU0MWU2MDJiYWUyMjZhNWY4YWE1Zjk1MjZkNDk3NjM0MDI4MWZh
+    NzgxZDllYzJkMzliZTkzMjQyY2ViY2I2ZTBmYzNkZDg2Y2FiMDc3OGJlMDkx
+    NjZkN2VmNjYyMWJkYTNlMDg2MjUxM2I3ZDYyZDVlOTc4Y2RiZTI=
+  data.tar.gz: !binary |-
+    OTA1ZTQ0MDA2M2MwOTZmNzBkMjY1MDBiNmY0M2E2ZTFkNTUyYzJhYzM5OTZk
+    MTg1ZDFkYjgyNDhmMDM1NmZlNWY2YmZjODBiNjA4MDQ5M2E3Njc1ZDA2Y2Nm
+    ODg0Njk1YWI4OTYwN2FmMTdhNzQxMWI3ZjFiYmU0MmUzMTQ4ZmY=

data/AUTHORS

@@ -0,0 +1 @@
+moaikids

data/Gemfile

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

data/README.rdoc

@@ -0,0 +1,50 @@
+= re-assembly streaming data (for Fluentd output plugin)
+
+== Overview
+
+
+== Installation
+
+== Configuration
+
+    <match pattern>
+      type reassemble
+      output_tag OUTPUT_TAG
+      assemble CONFIGURATION_FOR_REASSEMBLE
+      null_to_null true or false (default false)
+      null_to_empty true or false (default false)
+      datetime_format DATETIME_FORMAT (default '%Y/%m/%d %H:%M:%S' / used in operation = 'unixtime_to_datetime')
+      date_format DATE_FORMAT (default '%Y/%m/%d' / used in operation = 'unixtime_to_date')
+      time_format TIME_FORMAT (default '%H:%M:%S' / used in operation = 'unixtime_to_time')
+      tz TIMEZONE(default :local)
+    </match>
+
+=== 'assemble' format
+{extract_key1}:{replaced_key1}:{operation1},{extract_key2}:{replaced_key2}:{operation2},....{extract_keyN}:{replaced_keyN}:{operationN}
+
+ex1. 
+assemble foo1:bar1:to_i, foo2:bar2:to_s
+record => {"foo1": "1", "foo2": 2}
+reassemble => {"bar1": 1, "bar2": "2"}
+
+ex2. 
+assemble foo1,foo2:bar2,foo3:bar3:unixtime_to_datetime,foo4:bar4:url_to_host
+record => {"foo1": "1", "foo2": 2, "foo3": 1377946931, "foo4": "http://www.sada.co.jp/concert.html"}
+reassemble => {"foo1": "1", "bar2": 2, "bar3": "2013-08-31 20:00:02", "bar4": "www.sada.co.jp"}
+
+=== 'operation' list
+to_s    to string
+to_i    to integer
+to_f    to float
+unixtime_to_datetime    convert from unixtime to datetime(string)
+unixtime_to_date   convert from unixtime to date(string)
+unixtime_to_time   convert from unixtime to time(string)
+url_to_domain    extract host from url
+url_to_host    extract host from url
+url_to_path    extract path from url
+
+== Copyright
+
+Copyright:: Copyright (c) 2013 moaikids
+License::   Apache License, Version 2.0
+

data/Rakefile

@@ -0,0 +1,11 @@
+#!/usr/bin/env rake
+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-reassemble.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+Gem::Specification.new do |gem|
+    gem.name        = "fluent-plugin-reassemble"
+    gem.version     = "0.0.1"
+    gem.authors     = ["moaikids"]
+    gem.licenses    = ["Apache License Version 2.0"]
+    gem.summary     = %q{Re-assemble a stream data for Fluentd}
+    gem.description = %q{Re-assemble a stream data for Fluentd...}
+    gem.homepage    = "https://github.com/moaikids/fluent-plugin-reassemble"
+
+    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 "rake"
+    gem.add_development_dependency "shoulda"
+    gem.add_runtime_dependency "fluentd"
+end

data/lib/fluent/plugin/out_reassemble.rb

@@ -0,0 +1,119 @@
+require 'time'
+require 'uri'
+require 'tzinfo'
+
+module Fluent
+    class ReassembleOutput < Fluent::Output
+        Fluent::Plugin.register_output('reassemble', self)
+        config_param :output_tag, :string, :default => nil
+        config_param :assemble, :string, :default => nil
+        config_param :null_to_null, :bool, :default => false
+        config_param :null_to_empty, :bool, :default => false
+        config_param :datetime_format, :string, :default => '%Y/%m/%d %H:%M:%S'
+        config_param :date_format, :string, :default => '%Y/%m/%d'
+        config_param :time_format, :string, :default => '%H:%M:%S'
+        config_param :tz, :string, :default => nil
+
+        def configure(conf)
+            super
+            unless config.has_key?('output_tag')
+                raise Fluent::ConfigError, "you must set 'output_tag'"
+            end
+            unless config.has_key?('assemble')
+                raise Fluent::ConfigError, "you must set 'assemble'"
+            end
+            if config.has_key?('tz')
+                ENV["TZ"] = config['tz']
+            end
+            #assemble definition format
+            # {extract_key1}:{replaced_key1}:{operation1},{extract_key2}:{replaced_key2}:{operation2},....{extract_keyN}:{replaced_keyN}:{operationN}
+            @reassemble_conf = []
+            @assemble.split(",").each{ |conf|
+                extract, replace, operation = conf.split(":")
+                if extract.nil? || extract.empty?
+                    next
+                else
+                    extract = extract.strip
+                end
+                if replace.nil? || replace.empty?
+                    replace = extract
+                else
+                    replace = replace.strip
+                end
+                unless operation.nil? || operation.empty?
+                    operation = operation.strip
+                end
+                @reassemble_conf.push({:extract => extract, :replace => replace, :operation => operation})
+            }
+            $log.info "reassemble conf : " + @reassemble_conf.to_s
+        end
+
+        def emit(tag, es, chain)
+            es.each {|time,record|
+                json = {}
+                @reassemble_conf.each { |conf| 
+                    extract_key = conf[:extract]
+                    replaced_key = conf[:replace]
+                    operation = conf[:operation]
+                    val = convert(traverse(record, extract_key), operation)
+                    if val
+                        json[replaced_key] = val
+                    elsif @null_to_null
+                        json[replaced_key] = nil
+                    elsif @null_to_empty
+                        json[replaced_key] = ""
+                    end
+                }
+                Fluent::Engine.emit(@output_tag, time, json)
+            }
+        end 
+
+        def traverse(data, key)
+            val = data
+            key.split('.').each{ |k|
+                if val.is_a?(Hash) && val.has_key?(k)
+                    val = val[k]
+                else
+                    return nil
+                end
+            }
+            return val
+        end
+
+        def convert(val, operation)
+            unless val
+                return val
+            end
+            unless operation
+                return val
+            end
+         
+            begin
+                case operation.downcase
+                when "to_s"
+                    return val.to_s
+                when "to_i"
+                    return val.to_i
+                when "to_f"
+                    return val.to_f
+                when "unixtime_to_datetime"
+                    return Time.at(val.to_i).strftime(@datetime_format)
+                when "unixtime_to_date"
+                    return Time.at(val.to_i).strftime(@date_format)
+                when "unixtime_to_time"
+                    return Time.at(val.to_i).strftime(@time_format)
+                when "url_to_host","url_to_domain"
+                    return URI(val.to_s).host
+                when "url_to_path"
+                    return URI(val.to_s).path
+                else
+                    return val
+                end
+            rescue
+                $log.warn $!
+                return val
+            end
+
+        end
+    end
+end

metadata

@@ -0,0 +1,91 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-reassemble
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- moaikids
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-08-31 00:00:00.000000000 Z
+dependencies:
+- !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: shoulda
+  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: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Re-assemble a stream data for Fluentd...
+email: 
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- AUTHORS
+- Gemfile
+- README.rdoc
+- Rakefile
+- fluent-plugin-reassemble.gemspec
+- lib/fluent/plugin/out_reassemble.rb
+homepage: https://github.com/moaikids/fluent-plugin-reassemble
+licenses:
+- Apache License Version 2.0
+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.7
+signing_key: 
+specification_version: 4
+summary: Re-assemble a stream data for Fluentd
+test_files: []