put a Lock around self.client usage in call()

fetep · Mar 25, 2012 · 80e6929 · 80e6929 · tarekziade · Mar 25, 2012
1 parent aa51195
commit 80e6929
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/circus/flapping.py b/circus/flapping.py
@@ -1,5 +1,5 @@
 import errno
-from threading import Thread, Timer
+from threading import Lock, Thread, Timer
 import time
 import uuid
 
@@ -27,6 +27,7 @@ def __init__(self, context, endpoint, pubsub_endpoint, check_delay):
         self.timers = {}
         self.configs = {}
         self.tries = {}
+        self.zeromq_lock = Lock()
 
     @debuglog
     def initialize(self):
@@ -84,8 +85,10 @@ def handle_recv(self, data):
             self.update_conf(topic_parts[1])
 
     def call(self, cmd):
+        self.zeromq_lock.acquire()
         self.client.send(json.dumps(cmd))
         msg = self.client.recv()
+        self.zeromq_lock.release()
         return json.loads(msg)
 
     def update_conf(self, watcher_name):