fluent-plugin-dogstatsd 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 6ce5e12ea022b781f31f745f2a31ff3764860ab2
+  data.tar.gz: 83e3d99b0b172ac3a4e678e88299b2aaeb31257c
+SHA512:
+  metadata.gz: 779b3b7fbc3b189d88478edd560ee7dd65c02c8e3d29c02597f68dd11f3cee1ccb0139cd60a4317b2751566f5e23eedfe5959bb3ddd780ae7de8430db3edaf0d
+  data.tar.gz: 2f19c1bb877c1e22b581b39477221a4f0ff035e0b36fbccbd4cad556b6fb131053afd2177836e8358cbc464e84e6a8f3d3272fa3b4c4bf653cf5a0c1ae6bdc6b

data/.gitignore

@@ -0,0 +1,22 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Ryota Arai
+
+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,42 @@
+# fluent-plugin-dogstatsd
+
+Fluend plugin for Dogstatsd, that is statsd server for Datadog.
+
+## Installation
+
+    $ gem install fluent-plugin-dogstatsd
+
+## Usage
+
+```
+$ echo '{"type": "increment", "key": "apache.requests", "tags": {"url": "/"}}' | fluent-cat dogstatsd.hello
+$ echo '{"type": "histogram", "key": "apache.response_time", "value": 10.5, "tags": {"url": "/hello"}}' | fluent-cat dogstatsd.hello
+$ echo '{"type": "event", "title": "Deploy", "text": "New revision"}' | fluent-cat dogstatsd.hello
+```
+
+Supported types are `increment`, `decrement`, `count`, `gauge`, `histogram`, `timing`, `set` and `event`.
+
+## Configuration
+
+```
+<match dogstatsd.*>
+  type dogstatsd
+
+  # Dogstatsd host
+  host localhost
+
+  # Dogstatsd port
+  port 8125
+
+  # Use tag of fluentd record as key sent to Dogstatsd
+  use_tag_as_key false
+</match>
+```
+
+## Contributing
+
+1. Fork it ( https://github.com/ryotarai/fluent-plugin-dogstatsd/fork )
+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 a new Pull Request

data/Rakefile

@@ -0,0 +1,9 @@
+require "bundler/gem_tasks"
+
+require 'rake/testtask'
+
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'test'
+  test.test_files = FileList['test/plugin/*.rb']
+end
+

data/fluent-plugin-dogstatsd.gemspec

@@ -0,0 +1,25 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'fluent/plugin/dogstatsd/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "fluent-plugin-dogstatsd"
+  spec.version       = Fluent::Plugin::Dogstatsd::VERSION
+  spec.authors       = ["Ryota Arai"]
+  spec.email         = ["ryota.arai@gmail.com"]
+  spec.summary       = %q{Fluent plugin for Dogstatsd, that is statsd server for Datadog.}
+  spec.homepage      = "https://github.com/ryotarai/fluent-plugin-dogstatsd"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "fluentd"
+  spec.add_dependency "dogstatsd-ruby"
+
+  spec.add_development_dependency "bundler", "~> 1.6"
+  spec.add_development_dependency "rake"
+end

data/lib/fluent/plugin/dogstatsd.rb

@@ -0,0 +1,9 @@
+require "fluent/plugin/dogstatsd/version"
+
+module Fluent
+  module Plugin
+    module Dogstatsd
+      # Your code goes here...
+    end
+  end
+end

data/lib/fluent/plugin/dogstatsd/statsd_patch.rb

@@ -0,0 +1,9 @@
+require 'statsd'
+
+# https://github.com/DataDog/dogstatsd-ruby/pull/10
+class Statsd
+  def flush_buffer()
+    send_to_socket(@buffer.join("\n"))
+    @buffer = Array.new
+  end
+end

data/lib/fluent/plugin/dogstatsd/version.rb

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

data/lib/fluent/plugin/out_dogstatsd.rb

@@ -0,0 +1,77 @@
+module Fluent
+  class DogstatsdOutput < BufferedOutput
+    Plugin.register_output('dogstatsd', self)
+
+    config_param :host, :string, :default => nil
+    config_param :port, :integer, :default => nil
+    config_param :use_tag_as_key, :bool, :default => false
+
+    unless method_defined?(:log)
+      define_method(:log) { $log }
+    end
+
+    attr_accessor :statsd
+
+    def initialize
+      super
+
+      require 'statsd' # dogstatsd-ruby
+      require 'fluent/plugin/dogstatsd/statsd_patch'
+    end
+
+    def start
+      super
+
+      host = @host || Statsd::DEFAULT_HOST
+      port = @port || Statsd::DEFAULT_PORT
+
+      @statsd ||= Statsd.new(host, port)
+    end
+
+    def format(tag, time, record)
+      [tag, time, record].to_msgpack
+    end
+
+    def write(chunk)
+      @statsd.batch do |s|
+        chunk.msgpack_each do |tag, time, record|
+          key = if @use_tag_as_key
+                  tag
+                else
+                  record['key']
+                end
+
+          value = record['value']
+
+          options = {}
+
+          if record['tags']
+            options[:tags] = record['tags'].map do |k, v|
+              "#{k}:#{v}"
+            end
+          end
+
+          case record['type']
+          when 'increment'
+            s.increment(key, options)
+          when 'decrement'
+            s.decrement(key, options)
+          when 'count'
+            s.count(key, value, options)
+          when 'gauge'
+            s.gauge(key, value, options)
+          when 'histogram'
+            s.histogram(key, value, options)
+          when 'timing'
+            s.timing(key, value, options)
+          when 'set'
+            s.set(key, value, options)
+          when 'event'
+            s.event(record['title'], record['text'], options)
+          end
+        end
+      end
+    end
+  end
+end
+

data/test/plugin/test_out_dogstatsd.rb

@@ -0,0 +1,108 @@
+require 'test_helper'
+
+class DummyStatsd
+  attr_reader :messages
+
+  def initialize
+    @messages = []
+  end
+
+  def batch
+    yield(self)
+  end
+
+  %i!increment decrement count gauge histogram timing set event!.each do |name|
+    define_method(name) do |*args|
+      @messages << [name, args].flatten
+    end
+  end
+end
+
+class DogstatsdOutputTest < Test::Unit::TestCase
+  def setup
+    Fluent::Test.setup
+    require 'fluent/plugin/out_dogstatsd'
+  end
+
+  def teardown
+  end
+
+  def test_configure
+    d = create_driver(<<-EOC)
+      type dogstatsd
+      host HOST
+      port 12345
+    EOC
+
+    assert_equal('HOST', d.instance.host)
+    assert_equal(12345, d.instance.port)
+  end
+
+  def test_write
+    d = create_driver
+
+    d.instance.statsd = DummyStatsd.new
+
+    d.emit({'type' => 'increment', 'key' => 'hello.world1'}, Time.now.to_i)
+    d.emit({'type' => 'increment', 'key' => 'hello.world2'}, Time.now.to_i)
+    d.emit({'type' => 'decrement', 'key' => 'hello.world'}, Time.now.to_i)
+    d.emit({'type' => 'count', 'value' => 10, 'key' => 'hello.world'}, Time.now.to_i)
+    d.emit({'type' => 'gauge', 'value' => 10, 'key' => 'hello.world'}, Time.now.to_i)
+    d.emit({'type' => 'histogram', 'value' => 10, 'key' => 'hello.world'}, Time.now.to_i)
+    d.emit({'type' => 'timing', 'value' => 10, 'key' => 'hello.world'}, Time.now.to_i)
+    d.emit({'type' => 'set', 'value' => 10, 'key' => 'hello.world'}, Time.now.to_i)
+    d.emit({'type' => 'event', 'title' => 'Deploy', 'text' => 'Revision', 'key' => 'hello.world'}, Time.now.to_i)
+    d.run
+
+    assert_equal(d.instance.statsd.messages, [
+      [:increment, 'hello.world1', {}],
+      [:increment, 'hello.world2', {}],
+      [:decrement, 'hello.world', {}],
+      [:count, 'hello.world', 10, {}],
+      [:gauge, 'hello.world', 10, {}],
+      [:histogram, 'hello.world', 10, {}],
+      [:timing, 'hello.world', 10, {}],
+      [:set, 'hello.world', 10, {}],
+      [:event, 'Deploy', 'Revision', {}],
+    ])
+  end
+
+  def test_use_tag_as_key
+    d = create_driver(<<-EOC)
+#{default_config}
+use_tag_as_key true
+    EOC
+
+    d.instance.statsd = DummyStatsd.new
+
+    d.emit({'type' => 'increment'}, Time.now.to_i)
+    d.run
+
+    assert_equal(d.instance.statsd.messages, [
+      [:increment, 'dogstatsd.tag', {}],
+    ])
+  end
+
+  def test_tags
+    d = create_driver
+    d.instance.statsd = DummyStatsd.new
+    d.emit({'type' => 'increment', 'key' => 'hello.world', 'tags' => {'key' => 'value'}}, Time.now.to_i)
+    d.run
+
+    assert_equal(d.instance.statsd.messages, [
+      [:increment, 'hello.world', {tags: ["key:value"]}],
+    ])
+  end
+
+  private
+  def default_config
+    <<-EOC
+    type dogstatsd
+    EOC
+  end
+
+  def create_driver(conf = default_config)
+    Fluent::Test::BufferedOutputTestDriver.new(Fluent::DogstatsdOutput, 'dogstatsd.tag').configure(conf)
+  end
+end
+

data/test/test_helper.rb

@@ -0,0 +1,3 @@
+require 'test/unit'
+require 'fluent/test'
+

metadata

@@ -0,0 +1,114 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-dogstatsd
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Ryota Arai
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-09-09 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: dogstatsd-ruby
+  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: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !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'
+description: 
+email:
+- ryota.arai@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- fluent-plugin-dogstatsd.gemspec
+- lib/fluent/plugin/dogstatsd.rb
+- lib/fluent/plugin/dogstatsd/statsd_patch.rb
+- lib/fluent/plugin/dogstatsd/version.rb
+- lib/fluent/plugin/out_dogstatsd.rb
+- test/plugin/test_out_dogstatsd.rb
+- test/test_helper.rb
+homepage: https://github.com/ryotarai/fluent-plugin-dogstatsd
+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.2.2
+signing_key: 
+specification_version: 4
+summary: Fluent plugin for Dogstatsd, that is statsd server for Datadog.
+test_files:
+- test/plugin/test_out_dogstatsd.rb
+- test/test_helper.rb