tsd_client 0.0.1

data/lib/tsd_client/format.rb

@@ -0,0 +1,68 @@
+require 'uri'
+
+module TSD
+  module Format
+    def self.time input
+      case input.class.name
+      when 'Time'
+        input.strftime '%Y/%m/%d-%H:%M:%S'
+      when 'Fixnum'
+        time Time.at input
+      else
+        input
+      end
+    end
+
+    def self.tags tags
+      tags.collect do |tag, value|
+        [tag, value].join '='
+      end
+    end
+
+    def self.query_metric options
+      raise 'must provide key: :metric' unless options[:metric]
+      options = {aggregator: 'sum', rate: false}.merge options
+
+      metric_query = [options[:aggregator]]
+      metric_query << options[:downsample] if options[:downsample]
+      metric_query << 'rate' if options[:rate]
+
+      if options[:tags]
+        metric_query << [options[:metric], '{' + tags(options[:tags]).join(',') + '}'].join
+      else
+        metric_query << options[:metric]
+      end
+
+      metric_query.join ':'
+    end
+
+    def self.query options
+      raise 'must provide key: :start' unless options[:start]
+      options_filter = [:metric, :aggregator, :rate, :downsample, :tags]
+
+      # process query params
+      params = options.reduce({m: query_metric(options)}) do |params, (option, value)|
+        unless options_filter.include? option
+          params[option] = case option
+          when :start, :end # time formatting
+            time value
+          else
+            value
+          end        
+        end
+
+        params
+      end
+
+      # assemble query
+      URI.escape '/q?' + params.map {|option, value| [option, value].join('=')}.unshift('ascii').join('&')      
+    end
+
+    def self.put options
+      [:metric, :value].any? {|key| raise "missing key: #{key}" unless options.has_key? key}
+      options  = {time: Time.now, tags: {}}.merge options
+
+      ['put', options[:metric], options[:time].to_i, options[:value], tags(options[:tags])].flatten.join("\s")
+    end
+  end
+end

data/lib/tsd_client.rb

@@ -0,0 +1,49 @@
+$:.unshift File.join File.dirname(__FILE__), 'tsd_client'
+
+require 'socket'
+require 'timeout'
+require 'net/http'
+require 'tsd_client/format'
+
+module TSD
+  class Client
+    attr_reader :options
+
+    def initialize options
+      @options = {
+        host:    '0.0.0.0',
+        port:    4242,
+        timeout: 120, # seconds
+      }.merge options
+    end
+
+    def query options
+      response = nil
+
+      Timeout::timeout @options[:timeout] do
+        response = Net::HTTP.start @options[:host], @options[:port] do |http|
+          http.request Net::HTTP::Get.new Format.query options
+        end
+      end
+
+      if response.kind_of? Net::HTTPSuccess
+        response.body.split("\n").collect do |record|
+          metric, timestamp, value, *tags = record.split "\s"
+
+          Hash[[:metric, :time, :value, :tags].zip([
+            metric, Time.at(timestamp.to_i), value.to_f, Hash[tags.collect {|tag| tag.split('=')}]])]
+        end
+      else
+        raise 'query failed: ' + response.code
+      end
+    end
+
+    def put options
+      Timeout::timeout @options[:timeout] do
+        TCPSocket.open @options[:host], @options[:port] do |socket|
+          socket.puts Format.put options
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: tsd_client
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Dallas Marlow
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-03-24 00:00:00.000000000 Z
+dependencies: []
+description: 
+email:
+- dallasmarlow@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/tsd_client.rb
+- lib/tsd_client/format.rb
+homepage: https://github.com/dallasmarlow/tsd_client
+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: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: OpenTSDB client
+test_files: []
+has_rdoc: