Celery + FastAPI) Database Session 의존성 주입

Question

안녕하세요프레임워크는 FastAPI를 사용하고 있고,이번에 Celery를 이용해서 웹과는 별도로, 내부에서 작업 스케줄링을 걸려고 하고 있습니다.Celery 작업에서도 DB를 이용하는 작업이 있는데 문제가지금 Database Session 연결을 FastAPI의 Depends 의존성 주입 시켜놓아서, 정의된 라우터(URL path)로 요청이 들어올 때만 제너레이터(yield)를 이용해서 Session을 생성하는 방식으로 되어 있어서요# database.py def get_local_db(): db &#61; local_session() try: yield db finally: print('local_db closed.') db.close()# router.py &#64;router.post(&#34;/test&#34;) async def extract_rate(post_data: Request, db: Session &#61; Depends(get_bs_db)):Celery 작업은 API 요청과 관련 없이 동작시키고 싶은데, celery task 정의하는 곳에 인자로 Database Session을 주려하니, pickle 관련 에러 발생하면서 잘 안되더라고요...제가 생각해 본 방법은&#34;라우터을 추가해서, Celery 작업을 해당 URL로 요청을 보내도록 스케줄링 해놓는 방법&#34;(도 있겠지만, 위에서 말씀 드린 것처럼 별개로 동작시키고 싶고,,, 최후의 보루)Dependency Injector를 이용해서, 기존 Database 사용하는 곳에 의존성 주입을 따로 걸어놓는다?(해보려 햇는데 좀 복잡하고 잘 안되더라고요;;)이런 경우 어떻게 하는 게 좋은지 선배님들의 도움 부탁드립니다

Celery + FastAPI) Database Session 의존성 주입

0개의 답변

#celery 목록