Add the database primary key mode config (#953)

* Add the database primary key mode config

* Update auto to autoincrement
This commit is contained in:
Wu Clan
2025-12-05 14:52:12 +08:00
committed by GitHub
parent aad9afa9bb
commit bb5bcdf430
9 changed files with 39 additions and 27 deletions
+2 -1
View File
@@ -4,6 +4,7 @@ import celery
import celery_aio_pool
from backend.app.task.tasks.beat import LOCAL_BEAT_SCHEDULE
from backend.common.enums import DataBaseType
from backend.core.conf import settings
from backend.core.path_conf import BASE_PATH
@@ -32,7 +33,7 @@ def init_celery() -> celery.Celery:
broker_url = f'redis://:{settings.REDIS_PASSWORD}@{settings.REDIS_HOST}:{settings.REDIS_PORT}/{settings.CELERY_BROKER_REDIS_DATABASE}'
result_backend = f'db+postgresql+psycopg://{settings.DATABASE_USER}:{settings.DATABASE_PASSWORD}@{settings.DATABASE_HOST}:{settings.DATABASE_PORT}/{settings.DATABASE_SCHEMA}'
if settings.DATABASE_TYPE == 'mysql':
if DataBaseType.mysql == settings.DATABASE_TYPE:
result_backend = result_backend.replace('postgresql+psycopg', 'mysql+pymysql')
# https://docs.celeryq.dev/en/stable/userguide/configuration.html