[DOP-23620] Fix scheduler not closing DB connections

Author: dolfinus

syncmaster/scheduler/transfer_fetcher.py

@@ -4,7 +4,7 @@
 
 from syncmaster.db.models import Transfer
 from syncmaster.scheduler.settings import SchedulerAppSettings as Settings
-from syncmaster.scheduler.utils import get_async_session
+from syncmaster.scheduler.utils import get_async_engine, get_async_session
 
 
 class TransferFetcher:
@@ -13,8 +13,7 @@ def __init__(self, settings: Settings):
         self.last_updated_at = None
 
     async def fetch_updated_jobs(self) -> list[Transfer]:
-
-        async with get_async_session(self.settings) as session:
+        async with get_async_engine(self.settings) as engine, get_async_session(engine) as session:
             query = select(Transfer)
             if self.last_updated_at is not None:
                 query = query.filter(Transfer.updated_at > self.last_updated_at)

syncmaster/scheduler/transfer_job_manager.py

@@ -51,10 +51,9 @@ def update_jobs(self, transfers: list[Transfer]) -> None:
 
     async def remove_orphan_jobs(self) -> None:
         all_jobs = self.scheduler.get_jobs()
-        settings = self.settings
         job_transfer_ids = [int(job.id) for job in all_jobs]
 
-        async with get_async_engine(settings) as engine, get_async_session(engine) as session:
+        async with get_async_engine(self.settings) as engine, get_async_session(engine) as session:
             result = await session.execute(
                 select(Transfer).where(Transfer.id == any_(job_transfer_ids)),  # type: ignore[arg-type]
             )