bertclient 0.1 → 0.2

data/lib/bertclient.rb

@@ -13,13 +13,13 @@ module BERT
     class BadHeader < RPCError; end
     class BadData < RPCError; end
 
-    def initialize(opts={})
+    def initialize(opts={}, &block)
       @host = opts[:host] || 'localhost'
       @port = opts[:port] || 9999 
       @ssl = opts[:ssl] || false 
       @verify_ssl = opts.has_key?(:verify_ssl) ? opts[:verify_ssl] : true
       @socket = {}
-      connect
+      execute(&block) if block_given?
     end
 
     def call(mod, fun, *args)
@@ -75,7 +75,7 @@ module BERT
     end
 
     def socket
-      @socket[Thread.current]
+      @socket[Thread.current] ||= connect
     end
 
     # Open socket to service, use SSL if necessary
@@ -87,10 +87,23 @@ module BERT
         sock.connect
         sock.post_connection_check(@host) if @verify_ssl
       end
-      @socket[Thread.current] = sock
+      sock
+    end
+
+    # Close socket and clean it up from the pool
+    def close
+      socket.close
+      @socket.delete(Thread.current)
       true
     end
 
+    # Accepts a block, yields the client, closes socket at the end of the block
+    def execute
+      ret = yield self
+      close
+      ret
+    end
+
     # Reads a new berp from the socket and returns the decoded object
     def read_berp
       length = socket.read(4).unpack('N')[0]

metadata

@@ -1,7 +1,11 @@
 --- !ruby/object:Gem::Specification 
 name: bertclient
 version: !ruby/object:Gem::Version 
-  version: "0.1"
+  prerelease: false
+  segments: 
+  - 0
+  - 2
+  version: "0.2"
 platform: ruby
 authors: 
 - Jared Kuolt
@@ -9,7 +13,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-03-23 00:00:00 -07:00
+date: 2010-03-24 00:00:00 -07:00
 default_executable: 
 dependencies: []
 
@@ -36,18 +40,20 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
+      segments: 
+      - 0
       version: "0"
-  version: 
 required_rubygems_version: !ruby/object:Gem::Requirement 
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
+      segments: 
+      - 0
       version: "0"
-  version: 
 requirements: []
 
 rubyforge_project: bertclient
-rubygems_version: 1.3.5
+rubygems_version: 1.3.6
 signing_key: 
 specification_version: 3
 summary: A threadsafe BERT-RPC client with ssl support and persistent connections