fluent-plugin-amqp2 0.0.1

data/.gitignore

@@ -0,0 +1,3 @@
+Gemfile.lock
+.rvmrc
+*/pkg

data/Gemfile

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

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Restorando
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,52 @@
+# AMQP output plugin for Fluentd
+
+Fluentd output plugin to publish events to an amqp broker.
+
+Events are published one by one using the Fluentd tag as the routing key, in JSON format like:
+
+```javascript
+{ "key": "fluentd-tag", "timestamp": "fluentd-timestamp", "payload": "event-payload" }
+```
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'fluent-plugin-amqp2'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install fluent-plugin-amqp2
+
+## Configuration
+
+```
+<match pattern>
+  type amqp
+
+  # Set broker host and port
+  host localhost
+  port 5672
+
+  # Set user and password for authentication
+  user guest
+  password guest
+
+  # Configure amqp entities vhost, exchange id and type
+  vhost /
+  exchange my_exchange
+  exchange_type topic
+</match>
+```
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

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

data/fluent-plugin-amqp2.gemspec

@@ -0,0 +1,26 @@
+# -*- 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-amqp2"
+  gem.version       = "0.0.1"
+  gem.authors       = ["Augusto Becciu", "Juan Manuel Barreneche"]
+  gem.email         = ["devs@restorando.com"]
+
+  gem.description   = %q{AMQP output plugin for Fluent}
+  gem.summary       = %q{AMQP output plugin for Fluent}
+  gem.homepage      = "https://github.com/restorando/fluent-plugin-amqp"
+
+  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_dependency "fluentd", "~> 0.10.0"
+  gem.add_dependency "bunny", ">= 0.9.0.pre11"
+  gem.add_dependency "yajl-ruby", "~> 1.0"
+
+  gem.add_development_dependency "rake", ">= 0.9.2"
+  gem.add_development_dependency "mocha"
+end

data/lib/fluent/plugin/out_amqp.rb

@@ -0,0 +1,88 @@
+module Fluent
+
+class AmqpOutput < Fluent::BufferedOutput
+  # First, register the plugin. NAME is the name of this plugin
+  # and identifies the plugin in the configuration file.
+  Fluent::Plugin.register_output('amqp', self)
+
+  config_param :host, :string, default: "127.0.0.1"
+  config_param :port, :integer, default: 5672
+  config_param :user, :string, default: "guest"
+  config_param :password, :string, default: "guest"
+  config_param :vhost, :string, default: "/"
+  config_param :exchange, :string, default: ""
+  config_param :exchange_type, :string, default: "topic"
+
+  def initialize(*)
+    super
+    require "bunny"
+    require "yajl"
+  end
+
+  # This method is called before starting.
+  # 'conf' is a Hash that includes configuration parameters.
+  # If the configuration is invalid, raise Fluent::ConfigError.
+  def configure(conf)
+    super
+
+    raise Fluent::ConfigError, "missing host infromation" unless @host
+    raise Fluent::ConfigError, "missing exchange" unless @exchange
+
+    @exchange_name = @exchange
+  end
+
+  # This method is called when starting.
+  # Open sockets or files here.
+  def start
+    super
+
+    begin
+      get_or_create_exchange
+    rescue => e
+      $log.error "AMQP error", error: e.to_s
+      $log.warn_backtrace e.backtrace
+    end
+  end
+
+  # This method is called when shutting down.
+  # Shutdown the thread and close sockets or files here.
+  def shutdown
+    super
+    @amqp_conn && @amqp_conn.stop
+  end
+
+  def format(tag, time, record)
+    [tag, time, record].to_msgpack
+  end
+
+  # This method is called every flush interval. Write the buffer chunk
+  # to files or databases here.
+  # 'chunk' is a buffer chunk that includes multiple formatted
+  # events. You can use 'data = chunk.read' to get all events and
+  # 'chunk.open {|io| ... }' to get IO objects.
+  def write(chunk)
+    chunk.msgpack_each do |(tag, time, record)|
+      event = { "key" => tag, "timestamp" => time, "payload" => record }
+      get_or_create_exchange.publish Yajl.dump(event), routing_key: tag
+    end
+  end
+
+  private
+
+  def get_or_create_exchange
+    @amqp_conn ||= Bunny.new(host: @host, vhost: @vhost, port: @port,
+      user: @user, password: @password, threaded: false, automatically_recover: false)
+
+    @amqp_conn.start unless @amqp_conn.open?
+
+    unless @amqp_channel && @amqp_channel.open?
+      @amqp_channel  = @amqp_conn.create_channel
+      @amqp_exchange = Bunny::Exchange.new(@amqp_channel, @exchange_type.to_sym, @exchange_name, durable: true)
+    end
+
+    @amqp_exchange
+  end
+
+end
+
+end

data/test/out_amqp.rb

@@ -0,0 +1,85 @@
+require 'test/unit'
+require 'mocha/setup'
+
+require 'fluent/test'
+require 'fluent/plugin/out_amqp'
+
+class AmqpOutputTest < Test::Unit::TestCase
+
+  def setup
+    require 'bunny'
+    require 'yajl'
+    Fluent::Test.setup
+  end
+
+  CONFIG = %[
+    host localhost
+    port 3333
+    user test
+    password test
+    vhost /test
+    exchange test-exchange
+    exchange_type topic
+    buffert_type memory
+  ]
+
+  def create_driver(conf = CONFIG)
+    Fluent::Test::BufferedOutputTestDriver.new(Fluent::AmqpOutput).configure(conf)
+  end
+
+  def test_configure
+    d = create_driver
+    assert_equal "localhost", d.instance.host
+    assert_equal 3333, d.instance.port
+    assert_equal "test", d.instance.user
+    assert_equal "test", d.instance.password
+    assert_equal "/test", d.instance.vhost
+    assert_equal "test-exchange", d.instance.exchange
+    assert_equal "topic", d.instance.exchange_type
+  end
+
+  def test_start_and_shutdown
+    d = create_driver
+
+    amqp_conn_mock = mock()
+    amqp_exchange_mock = mock()
+    Bunny.stubs(:new).returns(amqp_conn_mock)
+
+    amqp_conn_mock.stubs(:open?).returns(false)
+    amqp_conn_mock.expects(:start)
+    amqp_conn_mock.expects(:stop)
+
+    amqp_conn_mock.expects(:create_channel)
+    Bunny::Exchange.expects(:new).returns(amqp_exchange_mock)
+
+    d.instance.start
+    # wait until thread starts
+    10.times { sleep 0.05 }
+    d.instance.shutdown
+  end
+
+  def test_flush
+    d = create_driver
+
+    amqp_conn_mock = mock()
+    amqp_exchange_mock = mock()
+    Bunny.stubs(:new).returns(amqp_conn_mock)
+    amqp_conn_mock.stubs(:open?).returns(true)
+    amqp_conn_mock.stubs(:create_channel)
+    amqp_conn_mock.stubs(:stop)
+    Bunny::Exchange.stubs(:new).returns(amqp_exchange_mock)
+
+    t = Time.now.to_i
+    d.emit({"a" => 1}, t)
+    d.emit({"a" => 2}, t)
+
+    ev1 = Yajl.dump({"key" => "test", "timestamp" => t, "payload" => {"a"=>1}})
+    ev2 = Yajl.dump({"key" => "test", "timestamp" => t, "payload" => {"a"=>2}})
+    amqp_exchange_mock.expects(:publish).with(ev1, { routing_key: "test" })
+    amqp_exchange_mock.expects(:publish).with(ev2, { routing_key: "test" })
+
+    d.run
+  end
+
+end
+

metadata

@@ -0,0 +1,141 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-amqp2
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Augusto Becciu
+- Juan Manuel Barreneche
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-27 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.10.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.10.0
+- !ruby/object:Gem::Dependency
+  name: bunny
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.9.0.pre11
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.9.0.pre11
+- !ruby/object:Gem::Dependency
+  name: yajl-ruby
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+- !ruby/object:Gem::Dependency
+  name: mocha
+  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'
+description: AMQP output plugin for Fluent
+email:
+- devs@restorando.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- fluent-plugin-amqp2.gemspec
+- lib/fluent/plugin/out_amqp.rb
+- test/out_amqp.rb
+homepage: https://github.com/restorando/fluent-plugin-amqp
+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'
+      segments:
+      - 0
+      hash: -856591938817169192
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -856591938817169192
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: AMQP output plugin for Fluent
+test_files:
+- test/out_amqp.rb