瀏覽代碼

[unrelated] Properly close storage on web app exit

jherve 1 年之前
父節點
當前提交
8c7d187433
共有 1 個文件被更改,包括 3 次插入1 次删除
  1. 3 1
      src/media_observer/web.py

+ 3 - 1
src/media_observer/web.py

@@ -71,7 +71,9 @@ async def get_db():
     if storage is None:
         storage = await Storage.create()
 
-    return storage
+    yield storage
+
+    await storage.close()
 
 
 async def get_similarity_search(storage: Storage = Depends(get_db)):