beanstalkd-stats 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 4252c3779ac2cfe56f7ab1e2f7c136c567c4303c
+  data.tar.gz: 79e87d48972aa2aefd947855c5ff074a42f1d0dc
+SHA512:
+  metadata.gz: 1cf63bfcbeb6479693af1e8be48cdc5af8a8079492ea02e7359b6b9c0b71fa3624f419e70d4652060ebdfb088cae93d5a7e71a3a6de2207c41dd735513d75a5c
+  data.tar.gz: 43387e5c169611f74254c5441bacce8730c4049df96fe509d59a43ba522fcbfffd0ad4a930b4d18d3a4950ea0c55625c0ab81c9321f2c85ad0e9d7bfba479078

data/bin/beanstalkd-stats

@@ -0,0 +1,73 @@
+#!/usr/bin/env ruby
+
+require 'open-uri'
+require 'net/http'
+require 'fileutils'
+require 'stackdriver'
+require 'beanstalk-client'
+require 'rufus-scheduler'
+require 'json'
+require 'optparse'
+require 'stat'
+
+options = {}
+optparse = OptionParser.new do |opts|
+    # todo: syntax
+    opts.on("-i", "--interval INTERVAL", "interval to query beanstalkd") do |interval|
+        options['interval'] = interval
+    end
+    opts.on("-h", "--host HOST", "beanstalkd host") do |host|
+        options['host'] = host
+    end
+    opts.on("-p", "--port PORT", "beanstalkd port") do |port|
+        options['port'] = port
+    end
+    opts.on("-c", "--config FILE", "config file" ) do |config_file|
+        file_json = File.read(config_file)
+        options.merge!(JSON.parse(file_json))
+    end
+    opts.on("-d", "--debug", "debug") do |debug|
+        options['debug'] = true
+    end
+end
+
+# Parsing arguments
+begin
+    optparse.parse!
+
+    mandatory = []
+    missing = mandatory.select{}
+rescue OptionParser::InvalidOption, OptionParser::MissingArgument
+    puts $!.to_s
+    puts optparse
+    exit 1
+end
+
+# StackDriver wants an instance_id
+def instance_id
+    output = IO.popen('/opt/aws/bin/ec2-metadata -i')
+    output.readlines.first.split[1]
+rescue
+    "localhost"
+end
+
+# begin
+server = "#{options['host']}:#{options['port']}"
+StackDriver.init options['stackdriver_api_key']
+scheduler = Rufus::Scheduler.new
+scheduler.every options['interval'] do
+    stats = []
+    connection = Beanstalk::Connection.new(server)
+    beanstalkd_stats = connection.stats
+
+    options['stats'].each do |m|
+        stat = Stat.new instance_id, m
+        stat.value = beanstalkd_stats[m] || 0
+        stats << stat.to_hash
+    end
+
+    puts stats.to_json if options['debug']
+    StackDriver.send_multi_metrics stats
+end
+
+scheduler.join

data/lib/stackdriver.rb

@@ -0,0 +1,64 @@
+require 'multi_json'
+
+module StackDriver
+  POST_URI = "https://custom-gateway.stackdriver.com/v1/custom"
+  DELETE_URI = "https://custom-gateway.stackdriver.com/v1/delete_custom"
+
+def self.init *args  
+  if args.count > 1
+    puts "Customer ID is no longer needed, and will be deprecated"
+    args.shift
+  end
+  @api_key = args[0]
+end
+
+  def self.send_metric name, value, time, instance=''
+    msg = build_message name, value, time, instance
+    post MultiJson.dump(msg), StackDriver::POST_URI
+  end
+
+  def self.send_multi_metrics data
+    msg = build_multi_message data
+    post MultiJson.dump(msg), StackDriver::POST_URI
+  end
+
+  def self.delete_metric name, time
+    msg = build_message name, nil, time
+    post MultiJson.dump(msg), StackDriver::DELETE_URI
+  end
+
+  private
+
+  def self.post msg, uri
+    headers = {'content-type' => 'application/json',
+      'x-stackdriver-apikey' => @api_key}
+
+    uri = URI(uri)
+    http = Net::HTTP.new(uri.host, uri.port)
+    http.use_ssl = true
+
+    http.start do |http|
+      response = http.post(uri.path, msg, headers)
+      if response.code != "201"
+        raise RuntimeError, "#{response.code} - #{response.body}"
+      end
+    end
+  end
+
+  def self.build_message name, value, time, instance=''
+    data_point = {'name' => name, 'value' => value, 'collected_at' => time}
+    data_point.merge!('value' => value) unless value.nil?
+    data_point.merge!('instance' => instance) unless instance.empty?
+    {'timestamp' => Time.now.to_i, 'proto_version' => '1', 'data' => data_point}
+  end
+
+  def self.build_multi_message data
+    data_point = data
+    {
+        'timestamp' => Time.now.to_i,
+        'proto_version' => '1',
+        'data' => data_point
+    }
+
+  end
+end

data/lib/stat.rb

@@ -0,0 +1,24 @@
+
+
+class Stat
+    attr_reader :instance, :name, :value, :collected_at
+
+    
+    def initialize(instance, name)
+        @instance=instance
+        @name=name
+        @value=0
+    end
+
+    def value=(newval)
+        @value=newval.to_i
+        @collected_at=Time.now
+    end
+
+    def to_hash
+        hash = {}
+        instance_variables.each {|var| hash[var.to_s.delete("@")] = instance_variable_get(var) }
+        hash
+    end
+end
+

metadata

@@ -0,0 +1,103 @@
+--- !ruby/object:Gem::Specification
+name: beanstalkd-stats
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Jorge Moratilla
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-10-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: multi_json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !ruby/object:Gem::Dependency
+  name: beanstalk-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: rufus-scheduler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.8'
+description: A simple beanstalkd gem
+email: jorge@moratilla.com
+executables:
+- beanstalkd-stats
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/beanstalkd-stats
+- lib/stackdriver.rb
+- lib/stat.rb
+homepage: http://rubygems.org/gems/beanstalkd-stats
+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: Stats for beanstalkd!
+test_files: []