feat(worker): 支持 WORKER_CONCURRENCY 环境变量控制 Celery 并发数 #208

Merged
xiaoxia merged 1 commits from feat/worker-concurrency-env into develop 2026-07-10 22:07:41 +08:00
2 changed files with 18 additions and 1 deletions
+13
View File
@@ -0,0 +1,13 @@
#!/bin/bash
# Worker 启动脚本 — 支持 WORKER_CONCURRENCY 环境变量
# 未设置时默认 2(保持向后兼容)
set -e
CONCURRENCY="${WORKER_CONCURRENCY:-2}"
exec celery \
-A worker_app.celery_app \
worker \
--loglevel=info \
"--concurrency=${CONCURRENCY}"
+5 -1
View File
@@ -48,6 +48,10 @@ COPY packages/ /app/packages/
COPY alembic.ini /app/alembic.ini
COPY migrations/ /app/migrations/
# 复制 Worker 启动脚本(支持 WORKER_CONCURRENCY 环境变量)
COPY infra/docker/entrypoint-worker.sh /usr/local/bin/entrypoint-worker.sh
RUN chmod +x /usr/local/bin/entrypoint-worker.sh
# 设置 Python 路径
ENV PATH="/opt/venv/bin:$PATH"
ENV PYTHONPATH=/app
@@ -61,4 +65,4 @@ USER celery
# Worker 入口点
WORKDIR /app/apps/worker
CMD ["celery", "-A", "worker_app.celery_app", "worker", "--loglevel=info", "--concurrency=2"]
CMD ["/usr/local/bin/entrypoint-worker.sh"]