fix: メモリ不足等でworkerが再起動した場合、dbの参照が消えてしまってエラーになるのを修正

This commit is contained in:
shimoda.m@nds-tyo.co.jp 2023-08-25 16:04:54 +09:00
parent ceb7e32de5
commit 6819e70704

View File

@ -5,4 +5,9 @@ from src.db.database import Database
def get_database(request: Request) -> Database:
# medaca_routerでDB接続エンジンが初期化される
return request.app.state._db
db = getattr(request.app.state, '_db', None)
# uvicornのワーカーが起動したタイミングでは、dbがセットされていないので、ここでセットここでセットする
if db is None:
db = Database.get_instance()
setattr(request.app.state, '_db', db)
return db