fluent-plugin-mongo-typed 0.1.0

data/.document

@@ -0,0 +1,5 @@
+lib/**/*.rb
+bin/*
+- 
+features/**/*.feature
+LICENSE.txt

data/Gemfile

@@ -0,0 +1,14 @@
+source "http://rubygems.org"
+
+gem "fluent-plugin-mongo", "~> 0.6.13", :require => false
+
+group :test do
+  gem "rspec", "~> 2.8.0"
+  gem "simplecov", "~> 0.7.1", :require => false
+end
+
+group :development do
+  gem "rdoc", "~> 3.12"
+  gem "bundler", "~> 1.0.0"
+  gem "jeweler", "~> 1.8.4"
+end

data/LICENSE.txt

@@ -0,0 +1,13 @@
+Copyright 2013 MORIYA Taro
+
+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.rdoc

@@ -0,0 +1,19 @@
+= fluent-plugin-mongo-typed
+
+Description goes here.
+
+== Contributing to fluent-plugin-mongo-typed
+ 
+* Check out the latest master to make sure the feature hasn't been implemented or the bug hasn't been fixed yet.
+* Check out the issue tracker to make sure someone already hasn't requested it and/or contributed it.
+* Fork the project.
+* Start a feature/bugfix branch.
+* Commit and push until you are happy with your contribution.
+* Make sure to add tests for it. This is important so I don't break it in a future version unintentionally.
+* Please try not to mess with the Rakefile, version, or history. If you want to have your own version, or is otherwise necessary, that is fine, but please isolate to its own commit so I can cherry-pick around it.
+
+== Copyright
+
+Copyright (c) 2013 MORIYA Taro. See LICENSE.txt for
+further details.
+

data/Rakefile

@@ -0,0 +1,45 @@
+# encoding: utf-8
+
+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 'rake'
+
+require 'jeweler'
+Jeweler::Tasks.new do |gem|
+  # gem is a Gem::Specification... see http://docs.rubygems.org/read/chapter/20 for more options
+  gem.name = "fluent-plugin-mongo-typed"
+  gem.homepage = "http://github.com/tarom/fluent-plugin-mongo-typed"
+  gem.license = "MIT"
+  gem.summary = %Q{Typed mongo output plugin}
+  gem.description = %Q{Typed mongo output plugin}
+  gem.email = "taro.toys@gmail.com"
+  gem.authors = ["MORIYA Taro"]
+  # dependencies defined in Gemfile
+  gem.add_dependency "fluent-plugin-mongo", "~> 0.6.13"
+end
+Jeweler::RubygemsDotOrgTasks.new
+
+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
+
+require 'rdoc/task'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "fluent-plugin-mongo-typed #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/fluent-plugin-mongo-typed.gemspec

@@ -0,0 +1,64 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{fluent-plugin-mongo-typed}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["MORIYA Taro"]
+  s.date = %q{2013-02-15}
+  s.description = %q{Typed mongo output plugin}
+  s.email = %q{taro.toys@gmail.com}
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+    "Gemfile",
+    "LICENSE.txt",
+    "README.rdoc",
+    "Rakefile",
+    "VERSION",
+    "fluent-plugin-mongo-typed.gemspec",
+    "lib/fluent/mixin/mongo_typed_mixin.rb",
+    "lib/fluent/plugin/out_mongo_replset_typed.rb",
+    "lib/fluent/plugin/out_mongo_typed.rb",
+    "test/helper.rb",
+    "test/plugin/test_out_mongo_replset_typed.rb",
+    "test/plugin/test_out_mongo_typed.rb"
+  ]
+  s.homepage = %q{http://github.com/tarom/fluent-plugin-mongo-typed}
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.6.0}
+  s.summary = %q{Typed mongo output plugin}
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<fluent-plugin-mongo>, ["~> 0.6.13"])
+      s.add_development_dependency(%q<rdoc>, ["~> 3.12"])
+      s.add_development_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.8.4"])
+      s.add_runtime_dependency(%q<fluent-plugin-mongo>, ["~> 0.6.13"])
+    else
+      s.add_dependency(%q<fluent-plugin-mongo>, ["~> 0.6.13"])
+      s.add_dependency(%q<rdoc>, ["~> 3.12"])
+      s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_dependency(%q<jeweler>, ["~> 1.8.4"])
+      s.add_dependency(%q<fluent-plugin-mongo>, ["~> 0.6.13"])
+    end
+  else
+    s.add_dependency(%q<fluent-plugin-mongo>, ["~> 0.6.13"])
+    s.add_dependency(%q<rdoc>, ["~> 3.12"])
+    s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+    s.add_dependency(%q<jeweler>, ["~> 1.8.4"])
+    s.add_dependency(%q<fluent-plugin-mongo>, ["~> 0.6.13"])
+  end
+end
+

data/lib/fluent/mixin/mongo_typed_mixin.rb

@@ -0,0 +1,53 @@
+module Fluent
+
+module MongoTypedMixin
+
+  ITEM_TYPES = ['string', 'integer', 'bool', 'time', 'array']
+
+  attr_accessor :item_types, :time_format
+
+  def configure(conf)
+    super
+
+    map = conf['item_types'].split(',').map do |type|
+      key, type = type.split(/:/)
+      if ITEM_TYPES.include?(type)
+        [key, type]
+      else
+        raise ConfigError, "mongo_ex: 'item_types' parameter format is \"KEY:TYPE,...\"\nTYPE is #{ITEM_TYPES.join(', ')}"
+      end
+    end
+    @item_types = Hash[*map.flatten(1)]
+    @time_format = conf['time_format']
+  end
+
+  def collect_records(chunk)
+    records = []
+    chunk.msgpack_each { |time, record|
+      record = Hash[*record.map{|key, value|cast(key, value)}.flatten(1)]
+      record[@time_key] = Time.at(time || record[@time_key]) if @include_time_key
+      records << record
+    }
+    records
+  end
+
+  def cast(key, value)
+    value = case @item_types[key]
+      when 'string'
+        value.to_s
+      when 'integer'
+        value.to_i
+      when 'bool'
+        Config.bool_value(value)
+      when 'time'
+        Time.strptime(value, @time_format)
+      when 'array'
+        value.split(/\s*,\s*/)
+      else
+        value
+      end
+    [key, value]
+  end
+end
+
+end

data/lib/fluent/plugin/out_mongo_replset_typed.rb

@@ -0,0 +1,11 @@
+module Fluent
+
+require 'fluent/plugin/out_mongo_replset'
+require 'fluent/mixin/mongo_typed_mixin'
+
+class MongoOutputReplsetTyped < MongoOutputReplset
+  Fluent::Plugin.register_output('mongo_replset_typed', self)
+  include MongoTypedMixin
+end
+
+end

data/lib/fluent/plugin/out_mongo_typed.rb

@@ -0,0 +1,11 @@
+module Fluent
+
+require 'fluent/plugin/out_mongo'
+require 'fluent/mixin/mongo_typed_mixin'
+
+class MongoOutputTyped < MongoOutput
+  Fluent::Plugin.register_output('mongo_typed', self)
+  include MongoTypedMixin
+end
+
+end

data/test/helper.rb

@@ -0,0 +1,20 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development, :test)
+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'
+require 'mongo'
+require 'fluent/plugin/out_mongo_typed'
+
+class Test::Unit::TestCase
+end

data/test/plugin/test_out_mongo_replset_typed.rb

@@ -0,0 +1,11 @@
+require 'helper'
+require 'plugin/test_out_mongo_typed'
+
+class TestMongoOutputReplsetTyped < TestMongoOutputTyped
+  DEFAULT_CONFIG = %[
+    type mongo_replset_typed
+    database #{DB_NAME}
+    collection #{COLLECTION_NAME}
+    nodes localhost
+  ]
+end

data/test/plugin/test_out_mongo_typed.rb

@@ -0,0 +1,64 @@
+require 'helper'
+
+class TestMongoOutputTyped < Test::Unit::TestCase
+
+  DB_NAME = '__fluent_test'
+  COLLECTION_NAME = 'test'
+  DEFAULT_CONFIG = %[
+    type mongo_typed
+    database #{DB_NAME}
+    collection #{COLLECTION_NAME}
+  ]
+
+  def setup
+    Fluent::Test.setup
+    @db = connect_db
+  end
+
+  def teardown
+    @db.collection(COLLECTION_NAME).drop
+    Mongo::Connection.new('localhost').drop_database(DB_NAME)
+  end
+
+  def connect_db
+    Mongo::Connection.new('localhost').db(DB_NAME)
+  end
+
+  def create_driver(conf = DEFAULT_CONFIG)
+    Fluent::Test::BufferedOutputTestDriver.new(Fluent::MongoOutputTyped).configure(conf)
+  end
+
+  def test_configure
+    time_format = "%d/%b/%Y:%H:%M:%S %z"
+    d = create_driver(DEFAULT_CONFIG + %[
+      item_types test1:integer,test2:string,test3:time,test4:bool
+      time_format #{time_format}
+    ]).instance
+
+    assert_equal('integer', d.item_types['test1'])
+    assert_equal('string', d.item_types['test2'])
+    assert_equal('time', d.item_types['test3'])
+    assert_equal('bool', d.item_types['test4'])
+
+    assert_equal(time_format, d.time_format)
+  end
+
+  def test_write
+    d = create_driver(DEFAULT_CONFIG + %[
+      item_types i:integer,s:string,t:time,b:bool,a:array
+      time_format %Y-%m-%d %H:%M:%S %z
+    ])
+    time = Time.parse('2013-02-12 22:01:15 UTC').to_i
+    t = '2013-02-12 22:04:14 UTC'
+    record = {'i' => '1', 's' => 'foo', 't' => t, 'b' => 'true', 'a' => 'a, b, c', 'o' => 'other'}
+    d.emit(record, time)
+    d.run
+    doc = @db.collection(COLLECTION_NAME).find_one()
+    assert_equal(1, doc['i'])
+    assert_equal('foo', doc['s'])
+    assert_equal(Time.gm(2013, 2, 12, 22, 4, 14), doc['t'])
+    assert_equal(true, doc['b'])
+    assert_equal(['a', 'b', 'c'], doc['a'])
+    assert_equal('other', doc['o'])
+  end
+end

metadata

@@ -0,0 +1,126 @@
+--- !ruby/object:Gem::Specification 
+name: fluent-plugin-mongo-typed
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.1.0
+platform: ruby
+authors: 
+- MORIYA Taro
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2013-02-15 00:00:00 +09:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: fluent-plugin-mongo
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 0.6.13
+  type: :runtime
+  prerelease: false
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rdoc
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: "3.12"
+  type: :development
+  prerelease: false
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: bundler
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 1.0.0
+  type: :development
+  prerelease: false
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: jeweler
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 1.8.4
+  type: :development
+  prerelease: false
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: fluent-plugin-mongo
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 0.6.13
+  type: :runtime
+  prerelease: false
+  version_requirements: *id005
+description: Typed mongo output plugin
+email: taro.toys@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE.txt
+- README.rdoc
+files: 
+- .document
+- Gemfile
+- LICENSE.txt
+- README.rdoc
+- Rakefile
+- VERSION
+- fluent-plugin-mongo-typed.gemspec
+- lib/fluent/mixin/mongo_typed_mixin.rb
+- lib/fluent/plugin/out_mongo_replset_typed.rb
+- lib/fluent/plugin/out_mongo_typed.rb
+- test/helper.rb
+- test/plugin/test_out_mongo_replset_typed.rb
+- test/plugin/test_out_mongo_typed.rb
+has_rdoc: true
+homepage: http://github.com/tarom/fluent-plugin-mongo-typed
+licenses: 
+- MIT
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 2373908222451003347
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.6.0
+signing_key: 
+specification_version: 3
+summary: Typed mongo output plugin
+test_files: []
+