Add asginl interface to skip ASGI lifespan (#153)

Author: gi0baro

README.md

@@ -82,7 +82,8 @@ Options:
                                   GRANIAN_HOST; default: 127.0.0.1]
   --port INTEGER                  Port to bind to.  [env var: GRANIAN_PORT;
                                   default: 8000]
-  --interface [asgi|rsgi|wsgi]    Application interface type  [env var:
+  --interface [asgi|asginl|rsgi|wsgi]
+                                  Application interface type  [env var:
                                   GRANIAN_INTERFACE; default: rsgi]
   --http [auto|1|2]               HTTP version  [env var: GRANIAN_HTTP;
                                   default: auto]

granian/asgi.py

@@ -33,7 +33,10 @@ async def handle(self):
             if self.failure_startup or self.failure_shutdown:
                 return
             self.unsupported = True
-            logger.warn('ASGI Lifespan errored, continuing without Lifespan support')
+            logger.warn(
+                'ASGI Lifespan errored, continuing without Lifespan support '
+                '(to avoid Lifespan completely use "asginl" interface)'
+            )
         finally:
             self.event_startup.set()
             self.event_shutdown.set()

granian/constants.py

@@ -3,6 +3,7 @@
 
 class Interfaces(str, Enum):
     ASGI = 'asgi'
+    ASGINL = 'asginl'
     RSGI = 'rsgi'
     WSGI = 'wsgi'
 

granian/server.py

@@ -164,6 +164,49 @@ def _spawn_asgi_worker(
             setproctitle.setproctitle(f'{process_name} worker-{worker_id}')
         configure_logging(log_level, log_config, log_enabled)
 
+        loop = loops.get(loop_impl)
+        sfd = socket.fileno()
+        callback = callback_loader()
+
+        shutdown_event = set_loop_signals(loop, [signal.SIGTERM, signal.SIGINT])
+
+        wcallback = _asgi_call_wrap(callback, scope_opts, {})
+        if not loop_opt:
+            wcallback = future_watcher_wrapper(wcallback)
+
+        worker = ASGIWorker(
+            worker_id, sfd, threads, pthreads, http_mode, http1_settings, http2_settings, websockets, loop_opt, *ssl_ctx
+        )
+        serve = getattr(worker, {ThreadModes.runtime: 'serve_rth', ThreadModes.workers: 'serve_wth'}[threading_mode])
+        serve(wcallback, loop, contextvars.copy_context(), shutdown_event)
+
+    @staticmethod
+    def _spawn_asgi_lifespan_worker(
+        worker_id,
+        process_name,
+        callback_loader,
+        socket,
+        loop_impl,
+        threads,
+        pthreads,
+        threading_mode,
+        http_mode,
+        http1_settings,
+        http2_settings,
+        websockets,
+        loop_opt,
+        log_enabled,
+        log_level,
+        log_config,
+        ssl_ctx,
+        scope_opts,
+    ):
+        from granian._loops import loops, set_loop_signals
+
+        if process_name:
+            setproctitle.setproctitle(f'{process_name} worker-{worker_id}')
+        configure_logging(log_level, log_config, log_enabled)
+
         loop = loops.get(loop_impl)
         sfd = socket.fileno()
         callback = callback_loader()
@@ -436,7 +479,8 @@ def serve(
         wrap_loader: bool = True,
     ):
         default_spawners = {
-            Interfaces.ASGI: self._spawn_asgi_worker,
+            Interfaces.ASGI: self._spawn_asgi_lifespan_worker,
+            Interfaces.ASGINL: self._spawn_asgi_worker,
             Interfaces.RSGI: self._spawn_rsgi_worker,
             Interfaces.WSGI: self._spawn_wsgi_worker,
         }