simple_metrics 0.2.3 → 0.3.2

data/bin/simple_metrics_client

@@ -1,64 +0,0 @@
-#!/usr/bin/env ruby
-
-require "rubygems"
-require "bundler/setup"
-
-require 'optparse'
-require "simple_metrics"
-
-options = {
-	:host        => 'localhost',
-	:port        => 8125,
-	:sample_rate => 1
-}
-
-parser ||= OptionParser.new do |opts|
-  opts.banner = "Usage Example: simple_metrics_send com.test.mymetric -c5"
-
-  opts.separator ""
-  opts.separator "Client options:"
-
-  opts.on("-c", "--counter VALUE", "Counter, a relative value") do |value|
-  	options[:type] = 'c'
-  	options[:stat] = value.to_i
-  end
-  opts.on("-g", "--gauge VALUE", "Gauge, an absolute value ") do |value|
-  	options[:type] = 'g'
-  	options[:stat] = value.to_i
-  end
-  opts.on("-t", "--timing VALUE", "A timing in ms") do |value|
-  	options[:type] = 'ms'
-  	options[:stat] = value.to_i
-  end
-  opts.on("-s", "--sample_rate VALUE", "An optional sample rate between 0 and 1 (example: 0.2)") do |value|
-  	options[:sample_rate] = value.to_f || 1
-  end
-
-  opts.separator ""
-
-  opts.on("-a", "--address HOST", "bind to HOST address (default: #{options[:host]})") do |host| 
-  	options[:host] = host
-  end
-
-  opts.on("-p", "--port PORT", "use PORT (default: #{options[:port]})") do |port|
-  	options[:port] = port.to_i
-  end
-
-  opts.separator ""
-  opts.on_tail("-h", "--help", "Show this message") { puts opts; exit }
-  opts.on_tail('-v', '--version', "Show version") { puts SimpleMetrics::VERSION; exit }
-
-end.parse!(ARGV)
-
-command    = ARGV.shift
-arguments  = ARGV
-client = SimpleMetrics::Client.new(options[:host])
-
-case options[:type]
-when'c' 
-	client.count(command, options[:stat], options[:sample_rate])
-when 'g'
-	client.gauge(command, options[:stat], options[:sample_rate])
-when 'ms'
-	client.timing(command, options[:stat], options[:sample_rate])
-end

data/lib/simple_metrics/client.rb

@@ -1,83 +0,0 @@
-# encoding: utf-8
-require "socket"
-
-module SimpleMetrics
-
-  class Client
-    VERSION = "0.0.1"
-    
-    def initialize(host, port = 8125)
-      @host, @port = host, port
-    end
-
-    # send relative value
-    def increment(stat, sample_rate = 1)
-      count(stat, 1, sample_rate)
-    end
-
-    # send relative value
-    def decrement(stat, sample_rate = 1)
-      count(stat, -1, sample_rate)
-    end
-
-    # send relative value
-    def count(stat, count, sample_rate = 1)
-      send_data( stat, count, 'c', sample_rate)
-    end
-
-    # send absolute value
-    # TODO: check if this is actually supported by Statsd server
-    def gauge(stat, value)
-      send_data(stat, value, 'g')
-    end
-
-    # Sends a timing (in ms) (glork)
-    def timing(stat, ms, sample_rate = 1)
-      send_data(stat, ms, 'ms', sample_rate)
-    end
-
-    # Sends a timing (in ms) block based
-    def time(stat, sample_rate = 1, &block)
-      start = Time.now
-      result = block.call
-      timing(stat, ((Time.now - start) * 1000).round, sample_rate)
-      result
-    end
-
-    private
-
-    def sampled(sample_rate, &block)
-      if sample_rate < 1
-        block.call if rand <= sample_rate
-      else
-        block.call
-      end
-    end
-
-    def send_data(stat, delta, type, sample_rate = 1)
-      sampled(sample_rate) do
-        data = "#{stat}:#{delta}|#{type}" # TODO: check stat is valid
-        data << "|@#{sample_rate}" if sample_rate < 1
-        send_to_socket(data)
-      end
-    end
-
-    def send_to_socket(data)
-      logger.debug "SimpleMetrics Client send: #{data}"
-      socket.send(data, 0, @host, @port)
-    rescue Exception => e
-      puts e.backtrace
-      logger.error "SimpleMetrics Client error: #{e}"
-    end
-
-    def socket
-      @socket ||= UDPSocket.new 
-    end
-
-    def logger
-      @logger ||= SimpleMetrics.logger
-    end
-
-  end
-
-end

data/lib/simple_metrics/mongo.rb

@@ -1,48 +0,0 @@
-# encoding: utf-8
-require "mongo"
-
-module SimpleMetrics
-  module Mongo
-    extend self
-
-    def ensure_collections_exist
-      SimpleMetrics.logger.debug "SERVER: MongoDB - found following collections: #{db.collection_names.inspect}"
-      Bucket.all.each do |bucket|
-        unless db.collection_names.include?(bucket.name)
-          db.create_collection(bucket.name, :capped => bucket.capped, :size => bucket.size) 
-          SimpleMetrics.logger.debug "SERVER: MongoDB - created collection #{bucket.name}, capped: #{bucket.capped}, size: #{bucket.size}"
-        end
-        
-        db.collection(bucket.name).ensure_index([['ts', ::Mongo::ASCENDING]])
-        SimpleMetrics.logger.debug "SERVER: MongoDB - ensure index on column ts for collection #{bucket.name}"
-      end 
-    end
-
-    def truncate_collections
-      Bucket.all.each do |bucket|
-        if db.collection_names.include?(bucket.name)
-          if bucket.capped?
-            collection(bucket.name).drop # capped collections can't remove elements, drop it instead
-          else
-            collection(bucket.name).remove
-          end
-          SimpleMetrics.logger.debug "SERVER: MongoDB - truncated collection #{bucket.name}"
-        end
-      end
-    end
-
-    @@collection = {}
-    def collection(name)
-      @@collection[name] ||= db.collection(name)
-    end
-
-    def connection
-      @@connection ||= ::Mongo::Connection.new(SimpleMetrics.db_config[:host], SimpleMetrics.db_config[:port])
-    end
-
-    def db
-      @@db ||= connection.db(SimpleMetrics.db_config[:db_name], SimpleMetrics.db_config[:options])
-    end
-
-  end
-end

data/lib/simple_metrics/server.rb

@@ -1,66 +0,0 @@
-# encoding: utf-8
-require "eventmachine"
-
-module SimpleMetrics
-
-  module ClientHandler
-
-    @@data_points = []
-
-    class << self
-      def get_and_clear_data_points
-        data_points = @@data_points.dup
-        @@data_points = []
-        data_points
-      end
-    end
-
-    def data_points
-      @@data_points
-    end
-
-    def post_init
-      SimpleMetrics.logger.info "ClientHandler entering post_init"
-    end
-
-    def receive_data(data)
-      SimpleMetrics.logger.debug "received_data: #{data.inspect}"
-
-      @@data_points ||= []
-      @@data_points << DataPoint.parse(data)
-    rescue DataPoint::ParserError => e
-      SimpleMetrics.logger.debug "Invalid Data skipped: #{data}"
-    end
-  end
-
-  class Server
-
-    attr_reader :db, :connection
-
-    def start
-      SimpleMetrics.logger.info "SERVER: starting up on #{SimpleMetrics.config[:host]}:#{SimpleMetrics.config[:port]}..."
-
-      Mongo.ensure_collections_exist
-
-      EM.run do
-        EM.open_datagram_socket(SimpleMetrics.config[:host], SimpleMetrics.config[:port], SimpleMetrics::ClientHandler) do |con|
-          EventMachine::add_periodic_timer(SimpleMetrics.config[:flush_interval]) do
-            SimpleMetrics.logger.debug "SERVER: period timer triggered after #{SimpleMetrics.config[:flush_interval]} seconds"
-
-            EM.defer { Bucket.flush_data_points(ClientHandler.get_and_clear_data_points) } 
-          end
-        end
-      end
-    end
-
-    def stop
-      SimpleMetrics.logger.info "EventMachine stop"
-      EM.stop
-    end
-    
-    def to_s
-      "#{SimpleMetrics.config[:host]}:#{SimpleMetrics.config[:port]}"
-    end
-
-  end
-end