add stop to the server

Author: naderzare

server.py

@@ -138,6 +138,7 @@ def serve(port):
     try:
         server.serve()
     except KeyboardInterrupt:
+        server.stop()
         print("Stopping server")
 
 

utils/PFProcessServer.py

@@ -9,6 +9,7 @@ class PFProcessServer(TServer):
 
     def __init__(self, *args):
         TServer.__init__(self, *args)
+        self.processors = []
         # self.processorFactory = args[0]
 
     def serve(self):
@@ -18,9 +19,8 @@ def serve(self):
                 client = self.serverTransport.accept()
                 if not client:
                     continue
-                p = multiprocessing.Process(target=self.handle, args=(client,))
-                p.start()
-
+                self.processors.append(multiprocessing.Process(target=self.handle, args=(client,)))
+                self.processors[-1].start()
             except KeyboardInterrupt:
                 raise
 
@@ -49,4 +49,9 @@ def handle(self, client):
         finally:
             itrans.close()
             if otrans:
-                otrans.close()
+                otrans.close()
+
+    def stop(self):
+        for p in self.processors:
+            p.terminate()
+        self.serverTransport.close()