rubycut-metriks 0.9.9.5 → 0.9.9.6

data/lib/metriks/reporter/opentsdb.rb

@@ -0,0 +1,109 @@
+require 'net/https'
+require 'socket'
+require 'io/wait'
+
+module Metriks::Reporter
+  class Opentsdb
+    attr_accessor :prefix, :source
+
+    def initialize(host, port, options = {})
+      @host = host
+      @port = port
+      @prefix = options[:prefix]
+      @source = options[:source]
+      @interval = options[:interval] || 60
+      @registry  = options[:registry] || Metriks::Registry.default
+      @on_error  = options[:on_error] || proc { |ex| }
+
+    end
+    def open_connection
+      @connection = TCPSocket.new(@host, @port)
+    end
+    def connection
+      @connection
+    end
+    def start
+      @thread ||= Thread.new do
+        loop do
+          sleep @interval
+          Thread.new do
+            begin
+              write
+            rescue Exception => ex
+              @on_error[ex] rescue nil
+            end
+          end
+        end
+      end
+    end
+
+    def stop
+      @thread.kill if @thread
+      @thread = nil
+    end
+
+    def restart
+      stop
+      start
+    end
+
+    def write
+      open_connection
+      @registry.each do |name, metric|
+        case metric
+        when Metriks::Meter
+          send_metric name, metric, [
+            :count, :one_minute_rate, :five_minute_rate,
+            :fifteen_minute_rate, :mean_rate
+          ]
+        when Metriks::Counter
+          send_metric name, metric, [
+            :count
+          ]
+        when Metriks::Gauge
+          send_metric name, metric, [
+            :value
+          ]
+        when Metriks::UtilizationTimer
+          send_metric name, metric, [
+            :count, :one_minute_rate, :five_minute_rate,
+            :fifteen_minute_rate, :mean_rate,
+            :min, :max, :mean, :stddev,
+            :one_minute_utilization, :five_minute_utilization,
+            :fifteen_minute_utilization, :mean_utilization,
+          ], [
+            :median, :get_95th_percentile
+          ]
+        when Metriks::Timer
+          send_metric name, metric, [
+            :count, :one_minute_rate, :five_minute_rate,
+            :fifteen_minute_rate, :mean_rate,
+            :min, :max, :mean, :stddev
+          ], [
+            :median, :get_95th_percentile
+          ]
+        when Metriks::Histogram
+          send_metric name, metric, [
+            :count, :min, :max, :mean, :stddev
+          ], [
+            :median, :get_95th_percentile
+          ]
+        end
+      end
+      close_connection
+
+    end
+    def close_connection
+      if connection.ready?
+       raise "Error while pushing data: #{connection.gets}"
+      end
+      connection.close
+    end
+    def send_metric(compound_name, metric, keys, snapshot_keys = [])
+      name, tags = compound_name.split("#")
+      keys.each do |key|
+        connection.puts("put #{name}.#{key} #{Time.now.to_i} #{metric.send(key)} #{tags}")
+      end
+    end
+  end
+end

data/metriks.gemspec

@@ -13,7 +13,7 @@ Gem::Specification.new do |s|
   ## If your rubyforge_project name is different, then edit it and comment out
   ## the sub! line in the Rakefile
   s.name              = 'rubycut-metriks'
-  s.version           = '0.9.9.5'
+  s.version           = '0.9.9.6'
   s.date              = '2013-04-28'
 
   ## Make sure your summary is short. The description may be as long
@@ -71,6 +71,7 @@ Gem::Specification.new do |s|
     lib/metriks/reporter/logger.rb
     lib/metriks/reporter/proc_title.rb
     lib/metriks/reporter/riemann.rb
+    lib/metriks/reporter/opentsdb.rb
     lib/metriks/simple_moving_average.rb
     lib/metriks/snapshot.rb
     lib/metriks/time_tracker.rb
@@ -92,6 +93,7 @@ Gem::Specification.new do |s|
     test/thread_error_handling_tests.rb
     test/timer_test.rb
     test/utilization_timer_test.rb
+    test/opentsdb_metrics_reporter_test.rb
   ]
   # = MANIFEST =
 

data/test/opentsdb_metrics_reporter_test.rb

@@ -0,0 +1,39 @@
+require 'test_helper'
+require 'thread_error_handling_tests'
+
+require 'metriks/reporter/opentsdb'
+
+class OpentsdbReporterTest < Test::Unit::TestCase
+  include ThreadErrorHandlingTests
+
+  def build_reporter(options={})
+    Metriks::Reporter::Opentsdb.new('localhost', 4242, { :registry => @registry }.merge(options))
+  end
+
+  def setup
+    @registry = Metriks::Registry.new
+    @reporter = build_reporter
+  end
+
+  def teardown
+    @reporter.stop
+    @registry.stop
+  end
+
+  def test_write
+    @registry.meter('meter.testing#tag=test').mark
+    @registry.counter('counter.testing#tag=test').increment
+    @registry.timer('timer.testing#tag=test').update(1.5)
+    @registry.histogram('histogram.testing#tag=test').update(1.5)
+    @registry.utilization_timer('utilization_timer.testing#tag=test').update(1.5)
+    @registry.gauge('gauge.testing#tag=test') { 123 }
+    tcp_socket = mock
+    @reporter.stubs(:connection).returns(tcp_socket)
+    tcp_socket.expects(:puts).at_least_once
+    tcp_socket.expects(:ready?).returns(false)
+    tcp_socket.expects(:close)
+
+    @reporter.connection.expects(:puts).with("put counter.testing.count #{Time.now.to_i} 1 tag=test")
+    @reporter.write
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification 
 name: rubycut-metriks
 version: !ruby/object:Gem::Version 
-  hash: 41
+  hash: 47
   prerelease: 
   segments: 
   - 0
   - 9
   - 9
-  - 5
-  version: 0.9.9.5
+  - 6
+  version: 0.9.9.6
 platform: ruby
 authors: 
 - Eric Lindvall
@@ -107,6 +107,7 @@ files:
 - lib/metriks/reporter/logger.rb
 - lib/metriks/reporter/proc_title.rb
 - lib/metriks/reporter/riemann.rb
+- lib/metriks/reporter/opentsdb.rb
 - lib/metriks/simple_moving_average.rb
 - lib/metriks/snapshot.rb
 - lib/metriks/time_tracker.rb
@@ -128,6 +129,7 @@ files:
 - test/thread_error_handling_tests.rb
 - test/timer_test.rb
 - test/utilization_timer_test.rb
+- test/opentsdb_metrics_reporter_test.rb
 homepage: https://github.com/rubycut/metriks
 licenses: []
 
@@ -174,4 +176,5 @@ test_files:
 - test/riemann_reporter_test.rb
 - test/timer_test.rb
 - test/utilization_timer_test.rb
+- test/opentsdb_metrics_reporter_test.rb
 has_rdoc: