Fix celery asynchronous task execution (#367)

backend/app/task/celery.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
-from celery import Celery
+import celery
+import celery_aio_pool
 
 from backend.app.task.conf import task_settings
 from backend.core.conf import settings
@@ -8,10 +9,21 @@ from backend.core.conf import settings
 __all__ = ['celery_app']
 
 
-def init_celery() -> Celery:
+def init_celery() -> celery.Celery:
     """创建 celery 应用"""
 
-    app = Celery('fba_celery')
+    # TODO: Update this work if celery version >= 6.0.0
+    # https://github.com/fastapi-practices/fastapi_best_architecture/issues/321
+    # https://github.com/celery/celery/issues/7874
+    celery.app.trace.build_tracer = celery_aio_pool.build_async_tracer
+    celery.app.trace.reset_worker_optimizations()
+
+    app = celery.Celery(
+        'fba_celery',
+        broker_connection_retry_on_startup=True,
+        worker_pool=celery_aio_pool.pool.AsyncIOPool,
+        trace=celery_aio_pool.build_async_tracer,
+    )
 
     # Celery Config
     # https://docs.celeryq.dev/en/stable/userguide/configuration.html

backend/app/task/celery_task/tasks.py

@@ -2,11 +2,14 @@
 # -*- coding: utf-8 -*-
 import uuid
 
+from anyio import sleep
+
 from backend.app.task.celery import celery_app
 
 
 @celery_app.task(name='task_demo_async')
-def task_demo_async() -> str:
+async def task_demo_async() -> str:
+    await sleep(1)
     uid = uuid.uuid4().hex
     print(f'异步任务 {uid} 执行成功')
     return uid