Listeners for main server process (#2018)

* Initial POC

* Add test case

* Resolve create_server and Gunicorn serve)

* add testr coverage

sanic/app.py

@@ -59,6 +59,7 @@ from sanic.server import (
     Signal,
     serve,
     serve_multiple,
+    serve_single,
 )
 from sanic.websocket import ConnectionClosed, WebSocketProtocol
 
@@ -817,7 +818,7 @@ class Sanic(BaseSanic):
                 )
                 workers = 1
             if workers == 1:
-                serve(**server_settings)
+                serve_single(server_settings)
             else:
                 serve_multiple(server_settings, workers)
         except BaseException:
@@ -920,6 +921,13 @@ class Sanic(BaseSanic):
             server_settings.get("before_start", []),
             server_settings.get("loop"),
         )
+        main_start = server_settings.pop("main_start", None)
+        main_stop = server_settings.pop("main_stop", None)
+        if main_start or main_stop:
+            logger.warning(
+                "Listener events for the main process are not available "
+                "with create_server()"
+            )
 
         return await serve(
             asyncio_server_kwargs=asyncio_server_kwargs, **server_settings
@@ -1038,6 +1046,8 @@ class Sanic(BaseSanic):
             ("after_server_start", "after_start", False),
             ("before_server_stop", "before_stop", True),
             ("after_server_stop", "after_stop", True),
+            ("main_process_start", "main_start", False),
+            ("main_process_stop", "main_stop", True),
         ):
             listeners = self.listeners[event_name].copy()
             if reverse:

sanic/mixins/listeners.py

@@ -13,6 +13,8 @@ class ListenerEvent(str, Enum):
     AFTER_SERVER_START = auto()
     BEFORE_SERVER_STOP = auto()
     AFTER_SERVER_STOP = auto()
+    MAIN_PROCESS_START = auto()
+    MAIN_PROCESS_STOP = auto()
 
 
 class ListenerMixin:

sanic/server.py

@@ -490,7 +490,7 @@ def serve(
                                   create_server method
     :return: Nothing
     """
-    if not run_async:
+    if not run_async and not loop:
         # create new event_loop after fork
         loop = asyncio.new_event_loop()
         asyncio.set_event_loop(loop)
@@ -603,7 +603,6 @@ def serve(
 
         trigger_events(after_stop, loop)
 
-        loop.close()
         remove_unix_socket(unix)
 
 
@@ -700,6 +699,23 @@ def remove_unix_socket(path: Optional[str]) -> None:
         pass
 
 
+def serve_single(server_settings):
+    main_start = server_settings.pop("main_start", None)
+    main_stop = server_settings.pop("main_stop", None)
+
+    if not server_settings.get("run_async"):
+        # create new event_loop after fork
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        server_settings["loop"] = loop
+
+    trigger_events(main_start, server_settings["loop"])
+    serve(**server_settings)
+    trigger_events(main_stop, server_settings["loop"])
+
+    server_settings["loop"].close()
+
+
 def serve_multiple(server_settings, workers):
     """Start multiple server processes simultaneously.  Stop on interrupt
     and terminate signals, and drain connections when complete.
@@ -712,6 +728,13 @@ def serve_multiple(server_settings, workers):
     server_settings["reuse_port"] = True
     server_settings["run_multiple"] = True
 
+    main_start = server_settings.pop("main_start", None)
+    main_stop = server_settings.pop("main_stop", None)
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+
+    trigger_events(main_start, loop)
+
     # Create a listening socket or use the one in settings
     sock = server_settings.get("sock")
     unix = server_settings["unix"]
@@ -752,5 +775,8 @@ def serve_multiple(server_settings, workers):
     for process in processes:
         process.terminate()
 
+    trigger_events(main_stop, loop)
+
     sock.close()
+    loop.close()
     remove_unix_socket(unix)

sanic/worker.py

@@ -7,6 +7,7 @@ import traceback
 
 from gunicorn.workers import base as base  # type: ignore
 
+from sanic.log import logger
 from sanic.server import HttpProtocol, Signal, serve, trigger_events
 from sanic.websocket import WebSocketProtocol
 
@@ -72,6 +73,15 @@ class GunicornWorker(base.Worker):
         )
         self._server_settings["before_start"] = ()
 
+        main_start = self._server_settings.pop("main_start", None)
+        main_stop = self._server_settings.pop("main_stop", None)
+
+        if main_start or main_stop:  # noqa
+            logger.warning(
+                "Listener events for the main process are not available "
+                "with GunicornWorker"
+            )
+
         self._runner = asyncio.ensure_future(self._run(), loop=self.loop)
         try:
             self.loop.run_until_complete(self._runner)