Speeding up bridge client. Fixes #4

Author: Federico Fissore

bridge/bridgeclient.py

@@ -30,6 +30,10 @@
 
 
 class BridgeClient:
+  def __init__(self):
+    self.json = None
+    self.should_close_at_function_end = True
+
   def wait_response(self, json, timeout):
     while timeout >= 0:
       r = json.recv()
@@ -50,39 +54,57 @@ def wait_key(self, key, json, timeout):
       except:
         pass
 
+  def socket_open(self):
+    if self.json is None:
+      self.json = TCPJSONClient('127.0.0.1', 5700)
+    return self.json
+
+  def socket_close(self):
+    if self.json is not None and self.should_close_at_function_end:
+      self.json.close()
+      self.json = None
+
+  def begin(self):
+    self.socket_open()
+    self.should_close_at_function_end = False
+
+  def close(self):
+    self.should_close_at_function_end = True
+    self.socket_close()
+
   def get(self, key):
-    json = TCPJSONClient('127.0.0.1', 5700)
+    json = self.socket_open()
     json.send({'command': 'get', 'key': key})
     r = self.wait_key(key, json, 10)
-    json.close()
+    self.socket_close()
     return r
 
   def getall(self):
-    json = TCPJSONClient('127.0.0.1', 5700)
+    json = self.socket_open()
     json.send({'command': 'get'})
     r = self.wait_response(json, 10)
     if not r is None:
       r = r['value']
-    json.close()
+    self.socket_close()
     return r
 
   def put(self, key, value):
-    json = TCPJSONClient('127.0.0.1', 5700)
+    json = self.socket_open()
     json.send({'command': 'put', 'key': key, 'value': value})
     r = self.wait_key(key, json, 10)
-    json.close()
+    self.socket_close()
     return r
 
   def delete(self, key):
-    json = TCPJSONClient('127.0.0.1', 5700)
+    json = self.socket_open()
     json.send({'command': 'delete', 'key': key})
     r = self.wait_response(json, 10)
     if not r is None and not r['value'] is None:
       r = r['value']
-    json.close()
+    self.socket_close()
     return r
 
   def mailbox(self, message):
-    json = TCPJSONClient('127.0.0.1', 5700)
+    json = self.socket_open()
     json.send({'command': 'raw', 'data': message})
-    json.close()
+    self.socket_close()

bridge/bridgeclient_example.py

@@ -39,5 +39,12 @@
 
 print
 
-print 'Sending mailbox message \'Hello world!\''
-client.mailbox('Hello world!')
+print 'Using client.begin() and client.close() to speed up consecutive calls to Bridge'
+
+client.begin()
+
+for idx in range(1, 51):
+  print 'Sending mailbox message \'Hello world ' + str(idx) + '!\''
+  client.mailbox('Hello world ' + str(idx) + '!')
+
+client.close()