feat: 添加统一启动脚本,支持通过环境变量配置workers数量
All checks were successful
continuous-integration/drone/push Build is passing
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:
@@ -1,64 +1,38 @@
|
||||
#!/bin/bash
|
||||
# 统一启动脚本 - 根据环境变量自动配置
|
||||
|
||||
# 颜色定义
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m' # No Color
|
||||
# 默认配置
|
||||
HOST=${HOST:-0.0.0.0}
|
||||
PORT=${PORT:-8000}
|
||||
WORKERS=${WORKERS:-1}
|
||||
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}"
|
||||
python_version=$(python3 --version 2>&1)
|
||||
if [[ $? -eq 0 ]]; then
|
||||
echo -e "${GREEN}✓ $python_version${NC}"
|
||||
# 构建启动命令
|
||||
CMD="uvicorn app.main:app --host $HOST --port $PORT --timeout-keep-alive $TIMEOUT_KEEP_ALIVE"
|
||||
|
||||
if [ "$RELOAD" = "true" ]; then
|
||||
# 开发模式:启用热重载(不支持多workers)
|
||||
CMD="$CMD --reload --reload-dir /app/app"
|
||||
echo "Mode: Development (hot reload enabled)"
|
||||
else
|
||||
echo -e "${RED}✗ Python3未安装${NC}"
|
||||
exit 1
|
||||
# 生产模式:多workers
|
||||
CMD="$CMD --workers $WORKERS"
|
||||
echo "Mode: Production ($WORKERS workers)"
|
||||
fi
|
||||
|
||||
# 检查虚拟环境
|
||||
if [ ! -d "venv" ]; then
|
||||
echo -e "${YELLOW}创建虚拟环境...${NC}"
|
||||
python3 -m venv venv
|
||||
fi
|
||||
echo ""
|
||||
echo "Executing: $CMD"
|
||||
echo ""
|
||||
|
||||
# 激活虚拟环境
|
||||
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
|
||||
exec $CMD
|
||||
|
||||
Reference in New Issue
Block a user