瀏覽代碼

fix celery backend when use rabbitmq (#23238)

Signed-off-by: kenwoodjw <blackxin55+@gmail.com>
tags/1.7.2
kenwoodjw 3 月之前
父節點
當前提交
1821726d4f
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 7 行新增2 行删除
  1. 7
    2
      api/configs/middleware/__init__.py

+ 7
- 2
api/configs/middleware/__init__.py 查看文件

@@ -215,7 +215,7 @@ class DatabaseConfig(BaseSettings):

class CeleryConfig(DatabaseConfig):
CELERY_BACKEND: str = Field(
description="Backend for Celery task results. Options: 'database', 'redis'.",
description="Backend for Celery task results. Options: 'database', 'redis', 'rabbitmq'.",
default="redis",
)

@@ -245,7 +245,12 @@ class CeleryConfig(DatabaseConfig):

@computed_field
def CELERY_RESULT_BACKEND(self) -> str | None:
return f"db+{self.SQLALCHEMY_DATABASE_URI}" if self.CELERY_BACKEND == "database" else self.CELERY_BROKER_URL
if self.CELERY_BACKEND in ("database", "rabbitmq"):
return f"db+{self.SQLALCHEMY_DATABASE_URI}"
elif self.CELERY_BACKEND == "redis":
return self.CELERY_BROKER_URL
else:
return None

@property
def BROKER_USE_SSL(self) -> bool:

Loading…
取消
儲存