add pause feature

samuelchen · samuelchen · commit edbe2787d704 · 2013-12-20T17:48:22.000+08:00
diff --git a/conn_mgr.py b/conn_mgr.py
@@ -96,6 +96,14 @@ def stop(self):
 
     def isAlive(self):
         return self.peerServer.isAlive() or self.dataServer.isAlive()
+    
+    @property
+    def paused(self):
+        return self.peerServer.paused
+
+    @paused.setter
+    def paused(self, val):
+        self.peerServer.paused = val
 
     def _broadcast(self, message=None, port=None, loop=False):
         '''
diff --git a/peer_server.py b/peer_server.py
@@ -25,6 +25,7 @@ class PeerServer(SocketServer.ThreadingMixIn, SocketServer.UDPServer):
     thread = None
     _heartbeat_loop = False
     _cast_loop = False
+    _paused = False
     
     assert(HEARTBEAT_LOOP >= BROADCAST_LOOP * 3)
 
@@ -87,6 +88,17 @@ def stop(self):
         else:
             self.log.info(':: PeerServer was shutdown.')
     
+    @property
+    def paused(self):
+        return self._paused
+
+    @paused.setter
+    def paused(self, val):
+        # TODO: expires caches if required
+        self._cast_loop = not val
+        self._heartbeat_loop = not val
+        self._paused = val
+        
     def isAlive(self):
         return self.thread and self.thread.isAlive()
 
