Add shutdown handling for Litestar

itssimon · itssimon · commit d98f653270a5 · 2024-03-06T22:03:56.000+10:00
diff --git a/apitally/litestar.py b/apitally/litestar.py
@@ -37,6 +37,7 @@ def __init__(
 
     def on_app_init(self, app_config: AppConfig) -> AppConfig:
         app_config.on_startup.append(self.on_startup)
+        app_config.on_shutdown.append(self.client.handle_shutdown)
         app_config.middleware.append(self.middleware_factory)
         return app_config
 
