From 2e05cc4b67473b6d5fd3f063bb976553dc00118c Mon Sep 17 00:00:00 2001
From: Bulat Kurbanov <kurbanovbul@gmail.com>
Date: Sat, 20 May 2023 20:38:57 +0200
Subject: [PATCH] Fix

---
 src/core/app.py | 41 +++++++++++++++++++----------------------
 1 file changed, 19 insertions(+), 22 deletions(-)

diff --git a/src/core/app.py b/src/core/app.py
index 70212fa..2e1b32a 100644
--- a/src/core/app.py
+++ b/src/core/app.py
@@ -1,5 +1,3 @@
-from contextlib import asynccontextmanager
-
 from fastapi import FastAPI
 from fastapi.responses import ORJSONResponse
 
@@ -12,33 +10,32 @@ from core.db import database
 from core.taskiq_worker import broker
 
 
-@asynccontextmanager
-async def lifespan(app: FastAPI):
-    if not database.is_connected:
-        await database.connect()
-
-    if not broker.is_worker_process:
-        await broker.startup()
-
-    yield
-
-    if database.is_connected:
-        await database.disconnect()
-
-    if not broker.is_worker_process:
-        await broker.shutdown()
-
-    await app.state.redis_pool.disconnect()
-
-
 def start_app() -> FastAPI:
-    app = FastAPI(default_response_class=ORJSONResponse, lifespan=lifespan)
+    app = FastAPI(default_response_class=ORJSONResponse)
 
     app.state.redis_pool = ConnectionPool.from_url(REDIS_URL)
 
     app.include_router(router)
     app.include_router(healthcheck_router)
 
+    @app.on_event("startup")
+    async def app_startup():
+        if not database.is_connected:
+            await database.connect()
+
+        if not broker.is_worker_process:
+            await broker.startup()
+
+    @app.on_event("shutdown")
+    async def app_shutdown():
+        if database.is_connected:
+            await database.disconnect()
+
+        if not broker.is_worker_process:
+            await broker.shutdown()
+
+        await app.state.redis_pool.disconnect()
+
     Instrumentator(
         should_ignore_untemplated=True,
         excluded_handlers=["/docs", "/metrics", "/healthcheck"],