kingkong 0.0.7 → 0.0.8

data/kingkong.gemspec

@@ -28,4 +28,6 @@ Gem::Specification.new do |s|
 
   s.add_runtime_dependency "eventmachine"
   s.add_runtime_dependency "nosey"
+  s.add_runtime_dependency "yajl-ruby"
+  s.add_runtime_dependency "em-http-request"
 end

data/lib/kingkong.rb

@@ -6,6 +6,7 @@ module KingKong
   autoload :Runner,     'kingkong/runner'
   autoload :Logging,    'kingkong/logging'
   autoload :Aggregator, 'kingkong/aggregator'
+  autoload :Processor,  'kingkong/processor'
 
   # Default logger for KingKong.
   def self.logger

data/lib/kingkong/ping.rb

@@ -70,7 +70,7 @@ module KingKong
     end
 
     def to_s
-      "Ping(#{id}, :#{status})"
+      "Ping(#{id}, :#{status}#{", #{latency}s" if completed?})"
     end
 
     # Generates ids for pings
@@ -83,6 +83,17 @@ module KingKong
       30 # 30 seconds that is!
     end
 
+    # Bust out a hash so that we can encode it into JSON and make some magic happen.
+    def to_hash
+      {
+        'status' => status,
+        'latency' => latency,
+        'start_time' => start_time,
+        'end_time' => end_time,
+        'ttl' => ttl
+      }
+    end
+
   private
     # Give us the current time in seconds
     def current_time

data/lib/kingkong/pinger.rb

@@ -1,11 +1,14 @@
 require 'eventmachine'
 require 'nosey'
+require 'em-http-request'
+require 'yajl'
+require 'time'
 
 module KingKong
+  # Executes pings within a specificed duration.
   class Pinger
     include Logging
-    include Nosey::Instrumentation
-
+    include EventMachine::Deferrable
     attr_reader :wait
 
     def initialize(wait=5,&block)
@@ -27,22 +30,24 @@ module KingKong
       @timer.cancel if @timer
     end
 
+    # Fire this if when a ping completes
+    def on_ping(&block)
+      @on_ping = block
+    end
+
   private
     # Add all of the instrumentation callbacks into the ping so we can aggregate it later
     def ping
       ping = Ping::Deferrable.new(Ping.default_ttl, sequencer)
-
       # Register the aggregator to process the ping
       ping.callback { 
-        logger.debug "Ping #{ping} successful"
-        process ping
+        logger.debug "Ping #{ping}"
+        @on_ping.call(ping) if @on_ping
       }
-
       ping.errback  {
-        logger.debug "Ping #{ping} error (probably a timeout)"
-        process ping
+        logger.debug "Ping #{ping}"
+        @on_ping.call(ping) if @on_ping
       }
-
       # Now pass the ping into the block so we can start/stop it
       @block.call(ping, self)
     end
@@ -51,19 +56,5 @@ module KingKong
     def sequencer
       @sequencer ||= Ping::Sequencer.new
     end
-
-    # Process a stinkin ping and report aggregate stats to Nosey
-    def process(ping)
-      nosey.increment 'ping_count'
-      case ping.status
-      when :timed_out
-        nosey.increment 'ping_timed_out_count'
-      when :completed
-        nosey.increment 'ping_completed_count'
-        nosey.avg "ping_avg_latency", ping.latency
-        nosey.min "ping_min_latency", ping.latency
-        nosey.max "ping_max_latency", ping.latency
-      end
-    end
   end
 end

data/lib/kingkong/processor.rb

@@ -0,0 +1,80 @@
+require 'nosey'
+
+module KingKong
+  module Processor
+    # Base class for processing pings
+    class Base
+      # Enable block configurations
+      def initialize(&block)
+        block.call(self) if block_given?
+        self
+      end
+
+      def process(ping)
+        raise 'Not Implemented'
+      end
+    end
+
+    class Nosey < Base
+      # include ::Nosey::Instrumentation
+
+      def initialize(host='/tmp/kingkong.socket',port=nil)
+        EventMachine::Nosey::SocketServer.start(nosey.report, host, port)
+      end
+
+      def process(ping,name)
+        nosey.increment "#{name}_ping_count"
+        case ping.status
+        when :timed_out
+          nosey.increment "#{name}_ping_timed_out_count"
+        when :completed
+          nosey.increment "#{name}_ping_completed_count"
+          nosey.avg "#{name}_ping_avg_latency", ping.latency
+          nosey.min "#{name}_ping_min_latency", ping.latency
+          nosey.max "#{name}_ping_max_latency", ping.latency
+        end
+      end
+      
+      def nosey
+        @nosey ||= ::Nosey::Probe::Set.new('pinger')
+      end
+    end
+
+    class Cube < Base
+      include Logging
+
+      attr_accessor :url
+      
+      # TODO use web sockets ...
+      def process(ping, name)
+        http = EM::HttpRequest.new(url).post({
+          :body => Yajl::Encoder.encode([cube_hash(ping, name)]),
+          :head => {
+            'content-type' => 'application/json'
+          }
+        })
+        http.callback{
+          case http.response_header.status
+          when 200..204
+            logger.debug "Successfully reported to Cube at #{url}"
+          else
+            logger.error "Could not report to Cube server: HTTP #{http.response_header.status} : #{http.response}"
+          end
+        }
+        http.errback{
+          logger.error "Could not connect to Cube at #{url}"
+        }
+      end
+
+    private
+    # Bust out a hash that cube will understand as JSON
+      def cube_hash(ping, name)
+        {
+          'type' => name.to_s,
+          'time' => Time.now.iso8601,
+          'data' => ping.to_hash
+        }
+      end
+    end
+  end
+end

data/lib/kingkong/runner.rb

@@ -8,19 +8,25 @@ module KingKong
     # Array of pingers that we're running
     def ping(name, &block)
       DSL::Pinger.new(&block).completed do |pinger|
-        pinger.nosey.name = name.to_s
+        pinger.on_ping do |ping|
+          process ping, name
+        end
         pingers << pinger
       end
     end
 
-    # Setup the socket that this thing will write stats out to
-    def socket(host='/tmp/king_kong.socket', port=nil)
-      @socket_host , @socket_port = host, port
+    # We use this method to pass our pings through a processor
+    def process(ping, name)
+      @processor.call(ping,name) if @processor
+    end
+
+    # Configure the processor that we use to report pings
+    def on_pong(&block)
+      @processor = block
     end
 
     # Start all of the pingers given the configurations
     def start
-      start_socket
       start_pingers
     end
 
@@ -50,18 +56,6 @@ module KingKong
     end
 
   private
-    # Fire up the reporting server if a socket (and port, optionally) are given
-    def start_socket
-      EM::Nosey::SocketServer.start(nosey_report, @socket_host, @socket_port) if @socket_host
-    end
-
-    # Get us the nosey report that our nosey socket needs to get the job done son!
-    def nosey_report
-      Nosey::Report.new do |r|
-        r.probe_sets = pingers.map(&:nosey)
-      end
-    end
-
     # Fire up all the pingers
     def start_pingers
       pingers.each(&:start)

data/lib/kingkong/version.rb

@@ -1,3 +1,3 @@
 module KingKong
-  VERSION = "0.0.7"
+  VERSION = "0.0.8"
 end

data/spec/lib/kingkong/ping_spec.rb

@@ -6,6 +6,10 @@ describe KingKong::Ping do
     KingKong::Ping.new.ttl.should eql(KingKong::Ping.default_ttl)
   end
 
+  it "should return hash" do
+    KingKong::Ping.new.to_hash.keys.should include(*%w[status latency start_time end_time ttl])
+  end
+
   context "active" do
     before(:each) do
       @ping = KingKong::Ping.new      # Open up a pinger

data/spec/lib/kingkong/pinger_spec.rb

@@ -1,29 +1,29 @@
-# require 'spec_helper'
+require 'spec_helper'
 
-# describe KingKong::Pinger do
-#   include EM::Ventually
+describe KingKong::Pinger do
+  include EM::Ventually
 
-#   before(:all) do
-#     @@pinged = false
-#     @pinger = KingKong::Pinger.new(0.1) do |ping, pinger|
-#       @@ping = ping
-#       ping.start
-#       ping.stop
-#       pinger.stop
-#     end
-#   end
+  before(:all) do
+    @@pinged = false
+    @pinger = KingKong::Pinger.new(0.1) do |ping, pinger|
+      @@ping = ping
+      ping.start
+      ping.stop
+      pinger.stop
+    end
+  end
 
-#   it "should ping" do
-#     @pinger.start
-#     ly(:completed) { @@ping.status }
-#   end
+  it "should ping" do
+    @pinger.start
+    ly(:completed) { @@ping.status }
+  end
 
-#   it "should stop"
-#   it "should start"
+  it "should stop"
+  it "should start"
 
-#   context "ping aggregation" do
-#     it "should process timed-out"
-#     it "should process successful"
-#     it "should process errors"
-#   end
-# end
+  context "ping aggregation" do
+    it "should process timed-out"
+    it "should process successful"
+    it "should process errors"
+  end
+end

data/spec/lib/kingkong/runner_spec.rb

@@ -8,8 +8,6 @@ describe KingKong::Runner do
     @@google_pinged = @@twitter_pinged = false
 
     @runner = KingKong::Runner.configure do |runner|
-      runner.socket '/tmp/king_kong_test.socket'
-
       runner.ping(:google).every(0.1).seconds do |ping|
         ping.start
         ping.stop
@@ -30,8 +28,16 @@ describe KingKong::Runner do
   end
 
   it "should write data to socket" do
+    nosey = KingKong::Processor::Nosey.new('/tmp/king_kong_test.socket')
+
+    @runner.on_pong do |ping, name|
+      nosey.process ping, name
+    end
+
     @runner.start
-    KingKong::Test::ReadSocket.start('/tmp/king_kong_test.socket').callback{|data|
+    c = KingKong::Test::ReadSocket.start('/tmp/king_kong_test.socket')
+    c.send_data("READ\nQUIT\n")
+    c.callback{|data|
       @data = data
     }
     ly{ @data }.test{|data| data =~ /max/ }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: kingkong
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.8
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,12 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-09-16 00:00:00.000000000Z
+date: 2011-09-21 00:00:00.000000000 -07:00
+default_executable: 
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70359584496540 !ruby/object:Gem::Requirement
+  requirement: &70240307047860 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +22,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70359584496540
+  version_requirements: *70240307047860
 - !ruby/object:Gem::Dependency
   name: guard-rspec
-  requirement: &70359584483820 !ruby/object:Gem::Requirement
+  requirement: &70240307047440 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +33,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70359584483820
+  version_requirements: *70240307047440
 - !ruby/object:Gem::Dependency
   name: growl
-  requirement: &70359584483400 !ruby/object:Gem::Requirement
+  requirement: &70240307047020 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +44,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70359584483400
+  version_requirements: *70240307047020
 - !ruby/object:Gem::Dependency
   name: rb-fsevent
-  requirement: &70359584482980 !ruby/object:Gem::Requirement
+  requirement: &70240307046600 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,10 +55,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70359584482980
+  version_requirements: *70240307046600
 - !ruby/object:Gem::Dependency
   name: em-ventually
-  requirement: &70359584482560 !ruby/object:Gem::Requirement
+  requirement: &70240307046180 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -65,10 +66,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70359584482560
+  version_requirements: *70240307046180
 - !ruby/object:Gem::Dependency
   name: timecop
-  requirement: &70359584482140 !ruby/object:Gem::Requirement
+  requirement: &70240311723120 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -76,10 +77,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70359584482140
+  version_requirements: *70240311723120
 - !ruby/object:Gem::Dependency
   name: ruby-debug19
-  requirement: &70359584481720 !ruby/object:Gem::Requirement
+  requirement: &70240311722700 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -87,10 +88,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70359584481720
+  version_requirements: *70240311722700
 - !ruby/object:Gem::Dependency
   name: eventmachine
-  requirement: &70359584481300 !ruby/object:Gem::Requirement
+  requirement: &70240311722280 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -98,10 +99,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70359584481300
+  version_requirements: *70240311722280
 - !ruby/object:Gem::Dependency
   name: nosey
-  requirement: &70359584480880 !ruby/object:Gem::Requirement
+  requirement: &70240311721860 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -109,16 +110,36 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70359584480880
+  version_requirements: *70240311721860
+- !ruby/object:Gem::Dependency
+  name: yajl-ruby
+  requirement: &70240311721440 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70240311721440
+- !ruby/object:Gem::Dependency
+  name: em-http-request
+  requirement: &70240311721020 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70240311721020
 description: Have you ever wanted to shoot a message throught Twitter, have your app
   pick it up, do some work on it, and report how long it takes? KingKong makes it
   slightly easier to do this with a DSL for writing custom pings and by providing
   basic reporting facilities that plug into graphing applications like Munin.
 email:
 - brad@bradgessler.com
-executables:
-- kingkong-munin
-- kingkong-server
+executables: []
 extensions: []
 extra_rdoc_files: []
 files:
@@ -128,14 +149,13 @@ files:
 - Guardfile
 - README.md
 - Rakefile
-- bin/kingkong-munin
-- bin/kingkong-server
 - kingkong.gemspec
 - lib/kingkong.rb
 - lib/kingkong/cli.rb
 - lib/kingkong/logging.rb
 - lib/kingkong/ping.rb
 - lib/kingkong/pinger.rb
+- lib/kingkong/processor.rb
 - lib/kingkong/runner.rb
 - lib/kingkong/version.rb
 - spec/lib/kingkong/ping_spec.rb
@@ -143,6 +163,7 @@ files:
 - spec/lib/kingkong/runner_spec.rb
 - spec/lib/kingkong_spec.rb
 - spec/spec_helper.rb
+has_rdoc: true
 homepage: ''
 licenses: []
 post_install_message: 
@@ -163,7 +184,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: kingkong
-rubygems_version: 1.8.10
+rubygems_version: 1.6.2
 signing_key: 
 specification_version: 3
 summary: Build complex network application health checks with Ruby and EventMachine

data/bin/kingkong-munin

@@ -1,18 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'socket'
-require 'yaml'
-
-puts case ARGV.first
- when /config/
-  %(graph_title Result latency
-graph_category App
-graph_vlabel load
-load.label load)  
-else
-  stats = YAML.load UNIXSocket.new("/tmp/kingkong.socket").gets("\r\n")
-  %(load.value #{stats[:avg]})
-end
-
-# TODO - Message throughput
-# TODO - Message latency (min/max/avg)

data/bin/kingkong-server

@@ -1,10 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'rubygems'
-require 'bundler/setup'
-require 'kingkong'
-
-EventMachine::run do
-  trap('TERM')  { EM.stop }
-  trap('INT')   { EM.stop }
-end