2025-12-17 11:02:42 +08:00
|
|
|
|
#!/bin/bash
|
|
|
|
|
|
set -e
|
|
|
|
|
|
|
|
|
|
|
|
echo "🚀 DeepAudit 后端启动中..."
|
|
|
|
|
|
|
|
|
|
|
|
# 等待 PostgreSQL 就绪
|
|
|
|
|
|
echo "⏳ 等待数据库连接..."
|
|
|
|
|
|
max_retries=30
|
|
|
|
|
|
retry_count=0
|
|
|
|
|
|
|
|
|
|
|
|
while [ $retry_count -lt $max_retries ]; do
|
|
|
|
|
|
if .venv/bin/python -c "
|
|
|
|
|
|
import asyncio
|
|
|
|
|
|
from sqlalchemy.ext.asyncio import create_async_engine
|
|
|
|
|
|
import os
|
|
|
|
|
|
|
|
|
|
|
|
async def check_db():
|
|
|
|
|
|
engine = create_async_engine(os.environ.get('DATABASE_URL', ''))
|
|
|
|
|
|
try:
|
|
|
|
|
|
async with engine.connect() as conn:
|
|
|
|
|
|
await conn.execute(text('SELECT 1'))
|
|
|
|
|
|
return True
|
|
|
|
|
|
except Exception:
|
|
|
|
|
|
return False
|
|
|
|
|
|
finally:
|
|
|
|
|
|
await engine.dispose()
|
|
|
|
|
|
|
|
|
|
|
|
from sqlalchemy import text
|
|
|
|
|
|
exit(0 if asyncio.run(check_db()) else 1)
|
|
|
|
|
|
" 2>/dev/null; then
|
|
|
|
|
|
echo "✅ 数据库连接成功"
|
|
|
|
|
|
break
|
|
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
|
|
retry_count=$((retry_count + 1))
|
|
|
|
|
|
echo " 重试 $retry_count/$max_retries..."
|
|
|
|
|
|
sleep 2
|
|
|
|
|
|
done
|
|
|
|
|
|
|
|
|
|
|
|
if [ $retry_count -eq $max_retries ]; then
|
|
|
|
|
|
echo "❌ 无法连接到数据库，请检查 DATABASE_URL 配置"
|
|
|
|
|
|
exit 1
|
|
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
|
|
# 运行数据库迁移
|
|
|
|
|
|
echo "📦 执行数据库迁移..."
|
|
|
|
|
|
.venv/bin/alembic upgrade head
|
|
|
|
|
|
|
|
|
|
|
|
echo "✅ 数据库迁移完成"
|
|
|
|
|
|
|
2026-01-08 17:00:13 +08:00
|
|
|
|
# 启动 gunicorn (多工作进程模式)
|
|
|
|
|
|
# 默认开启 4 个 worker，显著提高并发处理能力
|
|
|
|
|
|
echo "🌐 启动 API 服务 (Gunicorn + 4 Workers)..."
|
|
|
|
|
|
exec .venv/bin/gunicorn app.main:app \
|
|
|
|
|
|
--workers 4 \
|
|
|
|
|
|
--worker-class uvicorn.workers.UvicornWorker \
|
|
|
|
|
|
--bind 0.0.0.0:8000 \
|
|
|
|
|
|
--timeout 120 \
|
|
|
|
|
|
--access-logfile - \
|
|
|
|
|
|
--error-logfile -
|
