fluent-plugin-out_rawtcp 0.0.1

checksums.yaml

@@ -0,0 +1,15 @@
+---
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    ZjhmYWYzYTY1NzlmMTVhM2UwYmFjMzExMmZhOTgyZGQ4YThjN2M5Zg==
+  data.tar.gz: !binary |-
+    NjU5YmFlMGU4MDZhZjNlYTAwNmUzY2MwMDhmN2M5ZDgwOTk0ZDRjNw==
+SHA512:
+  metadata.gz: !binary |-
+    OTE3MDI4ZWE2NTQzZjI4NTgxNDE3MjBkMjgyMmIxODM1MGEwMGJjZjRiMmU2
+    MzM0MDkwNmQ3ZjcyZjgzZGRlZTExNWVlNjMwYTdlYjEyYjk0ODY3NDIyZjg2
+    NjE2YTJlZmI4YTIxMjg1MTUyZTU3NjdjOGJmNTU1Zjc2ZTAxNTQ=
+  data.tar.gz: !binary |-
+    MDE4YjM3NjU2ZDFhMjVkZGZmZjFkNjVhZjRlOWVmMzk3NzJlYzQ5NzYwYjA2
+    NTU5YmZhNTY0OGFmOWRiMjI5ZWM2YTgzYTFlMzYwODEwZWQ2ZDgyYjU4OTBj
+    YjhkMDNiMDNlMTc3YzYzMDc5YmJjNmUxZjViMDg2OTY0NzgwODE=

data/.gitignore

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

data/Gemfile

@@ -0,0 +1,2 @@
+source 'https://rubygems.org'
+gemspec

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/VERSION

@@ -0,0 +1 @@
+0.0.1

data/fluent-plugin-out_rawtcp.gemspec

@@ -0,0 +1,22 @@
+$:.push File.expand_path("../lib", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name          = "fluent-plugin-out_rawtcp"
+  s.version       = `cat VERSION`
+  s.authors       = ["lxfontes"]
+  s.email         = ["lxfontes+rawtcp@gmail.com"]
+  s.description   = %q{Raw tcp output plugin for Fluentd}
+  s.summary       = %q{output plugin for fluentd}
+  s.homepage      = "https://github.com/lxfontes/fluent-plugin-out_rawtcp"
+  s.license       = 'MIT'
+
+  s.files         = `git ls-files`.split($/)
+  s.executables   = s.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  s.test_files    = s.files.grep(%r{^(test|spec|features)/})
+  s.require_paths = ["lib"]
+
+  s.add_runtime_dependency "fluentd"
+
+  s.add_development_dependency "rake"
+  s.add_development_dependency "webmock"
+end

data/lib/fluent/plugin/out_rawtcp.rb

@@ -0,0 +1,125 @@
+# coding: utf-8
+module Fluent
+  class RawTcpOutput < BufferedOutput
+    Plugin.register_output('rawtcp', self)
+
+    def initialize
+      super
+      require 'socket'
+      require 'timeout'
+      require 'fileutils'
+      @nodes = []  #=> [Node]
+    end
+
+    config_param :send_timeout, :time, :default => 60
+    config_param :connect_timeout, :time, :default => 5
+    attr_reader :nodes
+
+    def configure(conf)
+      super
+
+      conf.elements.each do |e|
+        next if e.name != "server"
+
+        host = e['host']
+        port = e['port']
+        port = port ? port.to_i : DEFAULT_LISTEN_PORT
+
+        name = e['name']
+        unless name
+          name = "#{host}:#{port}"
+        end
+
+        node_conf = RawNodeConfig.new(name, host, port)
+        @nodes << Node.new(log, node_conf)
+        log.info "adding forwarding server '#{name}'", :host=>host, :port=>port
+      end
+    end
+
+    def start
+      super
+    end
+
+    def shutdown
+      super
+    end
+
+    def format(tag, time, record)
+      [tag, time, record].to_msgpack
+    end
+
+    def write(chunk)
+      return if chunk.empty?
+
+      error = nil
+
+      @nodes.each do |node|
+        begin
+          send_data(node, chunk)
+          return
+        rescue
+          error = $!
+        end
+      end
+
+      raise error if error
+      raise "No nodes available"
+    end
+
+    private
+    def send_data(node, chunk)
+      sock = connect(node)
+      begin
+        opt = [1, @send_timeout.to_i].pack('I!I!')  # { int l_onoff; int l_linger; }
+        sock.setsockopt(Socket::SOL_SOCKET, Socket::SO_LINGER, opt)
+
+        opt = [@send_timeout.to_i, 0].pack('L!L!')  # struct timeval
+        sock.setsockopt(Socket::SOL_SOCKET, Socket::SO_SNDTIMEO, opt)
+
+        chunk.msgpack_each do |tag, time, record|
+          next unless record.is_a? Hash
+          sock.write([tag, time, record].to_msgpack)
+        end
+      ensure
+        sock.close
+      end
+    end
+
+    def connect(node)
+      Timeout.timeout(@connect_timeout) do
+        return TCPSocket.new(node.resolved_host, node.port)
+      end
+    end
+
+    RawNodeConfig = Struct.new("RawNodeConfig", :name, :host, :port)
+
+    class Node
+      def initialize(log, conf)
+        @log = log
+        @conf = conf
+        @name = @conf.name
+        @host = @conf.host
+        @port = @conf.port
+        resolved_host  # check dns
+      end
+
+      attr_reader :conf
+      attr_reader :name, :host, :port
+      attr_accessor :failure, :available
+
+      def available?
+        @available
+      end
+
+      def standby?
+        @conf.standby
+      end
+
+      def resolved_host
+        @sockaddr = Socket.pack_sockaddr_in(@port, @host)
+        port, rhost = Socket.unpack_sockaddr_in(@sockaddr)
+        return rhost
+      end
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1 @@
+require 'minitest/pride'

data/test/plugin/test_out_rawtcp.rb

@@ -0,0 +1,41 @@
+require 'test/unit'
+
+require 'fluent/test'
+require 'fluent/plugin/out_rawtcp'
+
+require 'webmock/test_unit'
+require 'date'
+
+require 'helper'
+
+$:.push File.expand_path("../lib", __FILE__)
+$:.push File.dirname(__FILE__)
+
+WebMock.disable_net_connect!
+
+class TestRawTcpOutput < Test::Unit::TestCase
+  attr_accessor :index_cmds, :index_command_counts
+
+  def setup
+    Fluent::Test.setup
+    @driver = driver('test', "<server>
+                                      host 127.0.0.1
+                                      port 24225
+                              </server>")
+  end
+
+  def driver(tag='test', conf='')
+    @driver ||= Fluent::Test::BufferedOutputTestDriver.new(Fluent::RawTcpOutput, tag).configure(conf)
+  end
+
+  def sample_record
+    {'age' => 26, 'request_id' => '42', 'parent_id' => 'parent', 'sub' => {'field'=>{'pos'=>15}}}
+  end
+
+  def test_writes_to_default_index
+    100.times.each do |t|
+      @driver.emit(sample_record)
+    end
+    @driver.run
+  end
+end

metadata

@@ -0,0 +1,97 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-out_rawtcp
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- lxfontes
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-07-28 00:00:00.000000000 Z
+dependencies:
+- !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'
+- !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: webmock
+  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'
+description: Raw tcp output plugin for Fluentd
+email:
+- lxfontes+rawtcp@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- VERSION
+- fluent-plugin-out_rawtcp.gemspec
+- lib/fluent/plugin/out_rawtcp.rb
+- test/helper.rb
+- test/plugin/test_out_rawtcp.rb
+homepage: https://github.com/lxfontes/fluent-plugin-out_rawtcp
+licenses:
+- MIT
+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.3.0
+signing_key: 
+specification_version: 4
+summary: output plugin for fluentd
+test_files:
+- test/helper.rb
+- test/plugin/test_out_rawtcp.rb