bin/riemann-bench in riemann-tools-0.1.1 vs bin/riemann-bench in riemann-tools-0.1.2

@@ -1,21 +1,22 @@
 #!/usr/bin/env ruby
 
 # Connects to a server (first arg) and populates it with a constant stream of
 # events for testing.
 
+require 'rubygems'
 require 'riemann/client'
 require 'pp'
 
 class Riemann::Bench
   attr_accessor :client, :hosts, :services, :states
   def initialize
     @hosts = [nil] + (0...10).map { |i| "host#{i}" }
     @hosts = %w(a b c d e f g h i j)
     @services = %w(test1 test2 test3 foo bar baz xyzzy attack cat treat)
     @states = {}
-    @client = Riemann::Client.new(host: (ARGV.first || 'localhost'))
+    @client = Riemann::Client.new(:host => (ARGV.first || 'localhost'))
   end
 
   def evolve(state)
     m = state[:metric] + (rand - 0.5) * 0.1
     m = [[0,m].max, 1].min
@@ -28,15 +29,15 @@
     when 0.9..1.0
       'critical'
     end
 
     {
-      metric: m,
-      state: s,
-      host: state[:host],
-      service: state[:service],
-      description: "at #{Time.now}"
+      :metric => m,
+      :state => s,
+      :host => state[:host],
+      :service => state[:service],
+      :description => "at #{Time.now}"
     }
   end
   
   def tick
 #    pp @states
@@ -54,14 +55,14 @@
   end
 
   def start
     hosts.product(services).each do |host, service|
       states[[host, service]] = {
-        metric: 0.5,
-        state: 'ok',
-        description: "Starting up",
-        host: host,
-        service: service
+        :metric => 0.5,
+        :state => 'ok',
+        :description => "Starting up",
+        :host => host,
+        :service => service
       }
     end
   end
 end