fluent-plugin-zmq 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,8 @@
+source :rubygems
+
+# Specify your gem's dependencies in fluent-plugin-zmq.gemspec
+gemspec
+
+# fixme : make submodule
+gem "msgpack", "~>0.4.4"
+gem "fluentd" if RUBY_VERSION >= "1.9.2"

data/README

@@ -0,0 +1,3 @@
+Zeromq plugin for Fluent
+
+exit.

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/fluent-plugin-zmq.gemspec

@@ -0,0 +1,24 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "fluent-plugin-zmq/version"
+
+Gem::Specification.new do |s|
+  s.name        = "fluent-plugin-zmq"
+  s.version     = Fluent::Plugin::Zmq::VERSION
+  s.authors     = ["OZAWA Tsuyoshi"]
+  s.email       = ["ozawa.tsuyoshi@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{zmq plugin for fluent, an event collector}
+  s.description = %q{zmq plugin for fluent, an event collector}
+
+  s.rubyforge_project = "fluent-plugin-zmq"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  # s.add_runtime_dependency "rest-client"
+end

data/lib/fluent-plugin-zmq/version.rb

@@ -0,0 +1,7 @@
+module Fluent
+  module Plugin
+    module Zmq
+      VERSION = "0.0.1"
+    end
+  end
+end

data/lib/fluent/plugin/in_zmq.rb

@@ -0,0 +1,137 @@
+#!/usr/bin/env ruby
+
+require "fluent-plugin-zmq/version"
+#
+# Fluent
+#
+# Copyright (C) 2011 OZAWA Tsuyoshi
+#
+#    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.
+#
+
+
+module Fluent
+
+class ZMQInput < Input
+  Plugin.register_input('zmq', self)
+
+  config_param :port,            :integer, :default => 4010
+  config_param :bind,            :string,  :default => '0.0.0.0'
+  config_param :body_size_limit, :size,    :default => 32*1024*1024  # TODO default
+  #config_param :server_type,     :string,  :default => 'nonblocking'
+
+  def initialize
+    require 'zmq'
+    super
+  end
+
+  def configure(conf)
+    super
+  end
+
+  def start
+    Signal.trap(:INT){
+      puts ":INT"
+      @server.close
+      exit(0)
+    }
+
+    $log.debug "listening http on #{@bind}:#{@port}"
+    @zmq = ZMQ::Context.new
+    @server = @zmq.socket(ZMQ::UPSTREAM)
+    @server.bind("tcp://" + @bind + ":" + @port.to_s)
+    @thread = Thread.new(&method(:run))
+  end
+
+  def run
+    begin
+      while true
+        ret = ZMQ::select([@server])
+        ret[0].each do |sock|
+          msg = sock.recv
+          on_message(MessagePack.unpack(msg))
+        end
+      end
+    rescue
+      $log.error "unexpected error", :error=>$!.to_s
+      $log.error_backtrace
+    end
+  end
+
+
+  # message Entry {
+  #   1: long time
+  #   2: object record
+  # }
+  #
+  # message Forward {
+  #   1: string tag
+  #   2: list<Entry> entries
+  # }
+  #
+  # message PackedForward {
+  #   1: string tag
+  #   2: raw entries  # msgpack stream of Entry
+  # }
+  #
+  # message Message {
+  #   1: string tag
+  #   2: long? time
+  #   3: object record
+  # }
+  def on_message(msg)
+    # TODO format error
+    tag = msg[0].to_s
+    time = msg[1]
+    time = Engine.now if time == 0
+    record = msg[2]
+
+    p tag
+    p time
+    p record
+
+    Engine.emit(tag, time, record)
+
+    # TODO : support other formats
+    #if entries.class == String
+    #  # PackedForward
+    #  es = MessagePackEventStream.new(entries, @cached_unpacker)
+    #  Engine.emit_stream(tag, es)
+
+    #elsif entries.class == Array
+    #  # Forward
+    #  es = MultiEventStream.new
+    #  entries.each {|e|
+    #    time = e[0].to_i
+    #    time = (now ||= Engine.now) if time == 0
+    #    record = e[1]
+    #    es.add(time, record)
+    #  }
+    #  Engine.emit_stream(tag, es)
+
+    #else
+    #  # Message
+    #  time = msg[1]
+    #  time = Engine.now if time == 0
+    #  record = msg[2]
+    #  Engine.emit(tag, time, record)
+    #end
+  end
+
+  def shutdown
+    @server.close
+    #@thread.join # TODO
+  end
+end
+
+end

metadata

@@ -0,0 +1,52 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-zmq
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- OZAWA Tsuyoshi
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-10-30 00:00:00.000000000Z
+dependencies: []
+description: zmq plugin for fluent, an event collector
+email:
+- ozawa.tsuyoshi@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README
+- Rakefile
+- fluent-plugin-zmq.gemspec
+- lib/fluent-plugin-zmq/version.rb
+- lib/fluent/plugin/in_zmq.rb
+homepage: ''
+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-zmq
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: zmq plugin for fluent, an event collector
+test_files: []