#183 port open issue (#196)

sparckles · May 15, 2022 · b8fd373 · b8fd373
1 parent 8d7feb9
commit b8fd373
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/robyn/__init__.py b/robyn/__init__.py
@@ -104,6 +104,7 @@ def start(self, url="127.0.0.1", port=5000):
         """
 
         if not self.dev:
+            processes = []
             workers = self.workers
             socket = SocketHeld(url, port)
             for _ in range(self.processes):
@@ -122,8 +123,16 @@ def start(self, url="127.0.0.1", port=5000):
                     ),
                 )
                 p.start()
+                processes.append(p)
 
             print("Press Ctrl + C to stop \n")
+            try:
+                for process in processes:
+                    process.join()
+            except KeyboardInterrupt:
+                print(f"\n{Colors.BOLD}{Colors.OKGREEN} Terminating server!! {Colors.ENDC}")
+                for process in processes:
+                    process.kill()
         else:
             event_handler = EventHandler(self.file_path)
             event_handler.start_server_first_time()

diff --git a/robyn/processpool.py b/robyn/processpool.py
@@ -75,5 +75,9 @@ def spawn_process(
             web_socket.methods["message"],
         )
 
-    server.start(socket, workers)
-    asyncio.get_event_loop().run_forever()
+    try:
+        server.start(socket, workers)
+        loop = asyncio.get_event_loop()
+        loop.run_forever()
+    except KeyboardInterrupt:
+        loop.close()