feat: 添加统一启动脚本,支持通过环境变量配置workers数量
All checks were successful
continuous-integration/drone/push Build is passing

- 新增 start.sh 启动脚本,根据 WORKERS/RELOAD 环境变量自动配置
- 修改 Dockerfile 使用启动脚本,默认 WORKERS=4
- 更新 docker-compose.prod-multi.yml,所有租户使用环境变量配置
- 生产环境默认4个workers,提升并发处理能力

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
yuliang_guo
2026-02-03 14:25:34 +08:00
parent 879247c8e9
commit e110067840
3 changed files with 58 additions and 69 deletions

View File

@@ -43,9 +43,18 @@ RUN pip install --upgrade pip && \
# 复制应用代码 # 复制应用代码
COPY app/ ./app/ COPY app/ ./app/
# 复制启动脚本
COPY start.sh ./start.sh
RUN chmod +x ./start.sh
# 创建上传目录和日志目录 # 创建上传目录和日志目录
RUN mkdir -p uploads logs RUN mkdir -p uploads logs
# 默认环境变量可通过docker-compose或环境变量覆盖
ENV WORKERS=4 \
RELOAD=false \
TIMEOUT_KEEP_ALIVE=600
# 暴露端口 # 暴露端口
EXPOSE 8000 EXPOSE 8000
@@ -53,5 +62,5 @@ EXPOSE 8000
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \ HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
CMD curl -f http://localhost:8000/health || exit 1 CMD curl -f http://localhost:8000/health || exit 1
# 启动命令(生产模式,无热重载) # 使用启动脚本
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--workers", "4", "--timeout-keep-alive", "600"] CMD ["./start.sh"]

View File

@@ -1,64 +1,38 @@
#!/bin/bash #!/bin/bash
# 统一启动脚本 - 根据环境变量自动配置
# 颜色定义 # 默认配置
RED='\033[0;31m' HOST=${HOST:-0.0.0.0}
GREEN='\033[0;32m' PORT=${PORT:-8000}
YELLOW='\033[1;33m' WORKERS=${WORKERS:-1}
NC='\033[0m' # No Color RELOAD=${RELOAD:-false}
TIMEOUT_KEEP_ALIVE=${TIMEOUT_KEEP_ALIVE:-600}
echo -e "${GREEN}考培练系统后端启动脚本${NC}" echo "=============================================="
echo "================================" echo " KaoPeiLian Backend Starting..."
echo "=============================================="
echo " HOST: $HOST"
echo " PORT: $PORT"
echo " WORKERS: $WORKERS"
echo " RELOAD: $RELOAD"
echo " TIMEOUT_KEEP_ALIVE: $TIMEOUT_KEEP_ALIVE"
echo "=============================================="
# 检查Python版本 # 构建启动命令
echo -e "${YELLOW}检查Python版本...${NC}" CMD="uvicorn app.main:app --host $HOST --port $PORT --timeout-keep-alive $TIMEOUT_KEEP_ALIVE"
python_version=$(python3 --version 2>&1)
if [[ $? -eq 0 ]]; then if [ "$RELOAD" = "true" ]; then
echo -e "${GREEN}$python_version${NC}" # 开发模式启用热重载不支持多workers
CMD="$CMD --reload --reload-dir /app/app"
echo "Mode: Development (hot reload enabled)"
else else
echo -e "${RED}✗ Python3未安装${NC}" # 生产模式多workers
exit 1 CMD="$CMD --workers $WORKERS"
echo "Mode: Production ($WORKERS workers)"
fi fi
# 检查虚拟环境 echo ""
if [ ! -d "venv" ]; then echo "Executing: $CMD"
echo -e "${YELLOW}创建虚拟环境...${NC}" echo ""
python3 -m venv venv
fi
# 激活虚拟环境 exec $CMD
echo -e "${YELLOW}激活虚拟环境...${NC}"
source venv/bin/activate
# 安装依赖
echo -e "${YELLOW}安装依赖...${NC}"
pip install -q -r requirements/base.txt
# 检查.env文件
if [ ! -f ".env" ]; then
echo -e "${YELLOW}创建.env文件...${NC}"
cp .env.example .env
echo -e "${GREEN}✓ 已创建.env文件请根据需要修改配置${NC}"
fi
# 检查数据库连接
echo -e "${YELLOW}检查数据库连接...${NC}"
python -c "
import os
from dotenv import load_dotenv
load_dotenv()
db_url = os.getenv('DATABASE_URL', '')
if 'mysql' in db_url:
print('✓ 数据库配置已设置')
else:
print('⚠ 请检查数据库配置')
" 2>/dev/null
# 启动服务
echo -e "${GREEN}启动开发服务器...${NC}"
echo "================================"
echo -e "API文档: ${GREEN}http://localhost:8000/api/docs${NC}"
echo -e "健康检查: ${GREEN}http://localhost:8000/health${NC}"
echo "================================"
# 启动uvicorn
uvicorn app.main:app --reload --host 0.0.0.0 --port 8000

View File

@@ -78,16 +78,17 @@ services:
environment: environment:
- TZ=Asia/Shanghai - TZ=Asia/Shanghai
- PYTHONPATH=/app - PYTHONPATH=/app
- WORKERS=4 # 生产环境4个workers
- RELOAD=false # 生产环境关闭热重载
ports: ports:
- "8010:8000" - "8010:8000"
volumes: volumes:
- ./kaopeilian-backend/app:/app/app # 代码热重载 - ./kaopeilian-backend/app:/app/app
- /data/prod-envs/uploads-hua:/app/uploads - /data/prod-envs/uploads-hua:/app/uploads
- /data/prod-envs/logs-hua:/app/logs - /data/prod-envs/logs-hua:/app/logs
- /data/prod-envs/secrets:/app/secrets:ro - /data/prod-envs/secrets:/app/secrets:ro
- /etc/localtime:/etc/localtime:ro - /etc/localtime:/etc/localtime:ro
- /etc/timezone:/etc/timezone:ro - /etc/timezone:/etc/timezone:ro
command: ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]
networks: networks:
- prod-network - prod-network
- kaopeilian-network - kaopeilian-network
@@ -164,16 +165,17 @@ services:
environment: environment:
- TZ=Asia/Shanghai - TZ=Asia/Shanghai
- PYTHONPATH=/app - PYTHONPATH=/app
- WORKERS=4
- RELOAD=false
ports: ports:
- "8011:8000" - "8011:8000"
volumes: volumes:
- ./kaopeilian-backend/app:/app/app # 代码热重载 - ./kaopeilian-backend/app:/app/app
- /data/prod-envs/uploads-yy:/app/uploads - /data/prod-envs/uploads-yy:/app/uploads
- /data/prod-envs/logs-yy:/app/logs - /data/prod-envs/logs-yy:/app/logs
- /data/prod-envs/secrets:/app/secrets:ro - /data/prod-envs/secrets:/app/secrets:ro
- /etc/localtime:/etc/localtime:ro - /etc/localtime:/etc/localtime:ro
- /etc/timezone:/etc/timezone:ro - /etc/timezone:/etc/timezone:ro
command: ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]
networks: networks:
- prod-network - prod-network
- kaopeilian-network - kaopeilian-network
@@ -250,16 +252,17 @@ services:
environment: environment:
- TZ=Asia/Shanghai - TZ=Asia/Shanghai
- PYTHONPATH=/app - PYTHONPATH=/app
- WORKERS=4
- RELOAD=false
ports: ports:
- "8012:8000" - "8012:8000"
volumes: volumes:
- ./kaopeilian-backend/app:/app/app # 代码热重载 - ./kaopeilian-backend/app:/app/app
- /data/prod-envs/uploads-hl:/app/uploads - /data/prod-envs/uploads-hl:/app/uploads
- /data/prod-envs/logs-hl:/app/logs - /data/prod-envs/logs-hl:/app/logs
- /data/prod-envs/secrets:/app/secrets:ro - /data/prod-envs/secrets:/app/secrets:ro
- /etc/localtime:/etc/localtime:ro - /etc/localtime:/etc/localtime:ro
- /etc/timezone:/etc/timezone:ro - /etc/timezone:/etc/timezone:ro
command: ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]
networks: networks:
- prod-network - prod-network
- kaopeilian-network - kaopeilian-network
@@ -336,16 +339,17 @@ services:
environment: environment:
- TZ=Asia/Shanghai - TZ=Asia/Shanghai
- PYTHONPATH=/app - PYTHONPATH=/app
- WORKERS=4
- RELOAD=false
ports: ports:
- "8013:8000" - "8013:8000"
volumes: volumes:
- ./kaopeilian-backend/app:/app/app # 代码热重载 - ./kaopeilian-backend/app:/app/app
- /data/prod-envs/uploads-xy:/app/uploads - /data/prod-envs/uploads-xy:/app/uploads
- /data/prod-envs/logs-xy:/app/logs - /data/prod-envs/logs-xy:/app/logs
- /data/prod-envs/secrets:/app/secrets:ro - /data/prod-envs/secrets:/app/secrets:ro
- /etc/localtime:/etc/localtime:ro - /etc/localtime:/etc/localtime:ro
- /etc/timezone:/etc/timezone:ro - /etc/timezone:/etc/timezone:ro
command: ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]
networks: networks:
- prod-network - prod-network
- kaopeilian-network - kaopeilian-network
@@ -423,16 +427,17 @@ services:
environment: environment:
- TZ=Asia/Shanghai - TZ=Asia/Shanghai
- PYTHONPATH=/app - PYTHONPATH=/app
- WORKERS=4
- RELOAD=false
ports: ports:
- "8014:8000" - "8014:8000"
volumes: volumes:
- ./kaopeilian-backend/app:/app/app # 代码热重载 - ./kaopeilian-backend/app:/app/app
- /data/prod-envs/uploads-fw:/app/uploads - /data/prod-envs/uploads-fw:/app/uploads
- /data/prod-envs/logs-fw:/app/logs - /data/prod-envs/logs-fw:/app/logs
- /data/prod-envs/secrets:/app/secrets:ro - /data/prod-envs/secrets:/app/secrets:ro
- /etc/localtime:/etc/localtime:ro - /etc/localtime:/etc/localtime:ro
- /etc/timezone:/etc/timezone:ro - /etc/timezone:/etc/timezone:ro
command: ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]
networks: networks:
- prod-network - prod-network
- kaopeilian-network - kaopeilian-network
@@ -508,6 +513,8 @@ services:
environment: environment:
- TZ=Asia/Shanghai - TZ=Asia/Shanghai
- PYTHONPATH=/app - PYTHONPATH=/app
- WORKERS=4
- RELOAD=false
ports: ports:
- "8016:8000" - "8016:8000"
volumes: volumes:
@@ -517,7 +524,6 @@ services:
- /data/prod-envs/secrets:/app/secrets:ro - /data/prod-envs/secrets:/app/secrets:ro
- /etc/localtime:/etc/localtime:ro - /etc/localtime:/etc/localtime:ro
- /etc/timezone:/etc/timezone:ro - /etc/timezone:/etc/timezone:ro
command: ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]
networks: networks:
- prod-network - prod-network
- kaopeilian-network - kaopeilian-network