beetle_joevandyk_extensions 0.0.2 → 0.0.3

data/lib/beetle_joevandyk_extensions.rb

@@ -1,16 +1,23 @@
 require "beetle_joevandyk_extensions/version"
 require 'beetle'
+require 'timeout'
 
 module BeetleJoevandykExtensions
+  class BeetleError < RuntimeError; end
+
+  DEFAULT_TIMEOUT = 10
+
   def rpc name, message, options={}
     client = Beetle::Client.new
     client.register_message name
-    status, result = client.rpc(name, message.to_json)
-    raise result.to_s if status != "OK"
-    JSON.parse(result)
+    Timeout.timeout(options[:timeout] || DEFAULT_TIMEOUT, BeetleError) do
+      status, result = client.rpc(name, message.to_json)
+      raise BeetleError.new(result.to_s) if status != "OK"
+      JSON.parse(result)
+    end
   end
 
-  def listener name, &block
+  def listen name, &block
     client = Beetle::Client.new
     client.register_queue name
     client.register_handler name do |input|
@@ -20,6 +27,12 @@ module BeetleJoevandykExtensions
       puts "Started #{ name } at #{ Time.now }"
     end
   end
+
+  def publish name, message
+    client = Beetle::Client.new
+    client.register_message name
+    client.publish name, message
+  end
 end
 
 Beetle.extend(BeetleJoevandykExtensions)

data/lib/beetle_joevandyk_extensions/version.rb

@@ -1,3 +1,3 @@
 module BeetleJoevandykExtensions
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: beetle_joevandyk_extensions
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors: