fix: Close db connection on startup

bihealth · Nov 15, 2023 · 596ea55 · 596ea55
1 parent 0ff3d21
commit 596ea55
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/backend/app/backend_pre_start.py b/backend/app/backend_pre_start.py
@@ -23,15 +23,15 @@
 )
 async def init():
     try:
-        db = SessionLocal()
-        # Try to create session to check if DB is awake
-        await db.execute(text("SELECT 1"))
-        # Ensure to run Alembic on startup
-        for key in ("DATABASE_URL", "SQLALCHEMY_DATABASE_URI"):
-            if key in os.environ:
-                os.environ[key] = os.environ[key].replace("+asyncpg", "")
-        alembicArgs = ["--raiseerr", "upgrade", "head"]
-        alembic.config.main(alembicArgs)
+        with SessionLocal() as db:
+            # Try to create session to check if DB is awake
+            await db.execute(text("SELECT 1"))
+            # Ensure to run Alembic on startup
+            for key in ("DATABASE_URL", "SQLALCHEMY_DATABASE_URI"):
+                if key in os.environ:
+                    os.environ[key] = os.environ[key].replace("+asyncpg", "")
+            alembicArgs = ["--raiseerr", "upgrade", "head"]
+            alembic.config.main(alembicArgs)
     except Exception as e:
         logger.error(e)
         raise e