Codex
62 lines
1.7 KiB
Bash
62 lines
1.7 KiB
Bash
#!/bin/bash
|
|
set -e
|
|
|
|
echo "=========================================="
|
|
echo " ZQ Platform Backend - Docker Entrypoint"
|
|
echo "=========================================="
|
|
|
|
echo "[1/4] 等待数据库连接..."
|
|
if [ -n "$DB_HOST" ] && [ -n "$DB_PORT" ]; then
|
|
/wait-for-it.sh "$DB_HOST" "$DB_PORT" 60
|
|
fi
|
|
|
|
echo "[2/4] 等待 Redis 连接..."
|
|
if [ -n "$REDIS_HOST" ] && [ -n "$REDIS_PORT" ]; then
|
|
/wait-for-it.sh "$REDIS_HOST" "$REDIS_PORT" 30
|
|
fi
|
|
|
|
echo "[3/4] 数据库初始化..."
|
|
|
|
MIGRATION_COUNT=$(ls alembic/versions/*.py 2>/dev/null | wc -l)
|
|
|
|
if [ "$MIGRATION_COUNT" -eq "0" ]; then
|
|
echo " 未发现迁移文件，自动生成..."
|
|
python -m alembic revision --autogenerate -m "initial" 2>&1 | tail -3
|
|
fi
|
|
|
|
python -m alembic upgrade head 2>&1 || {
|
|
echo " [WARN] 迁移失败，5秒后重试..."
|
|
sleep 5
|
|
python -m alembic upgrade head 2>&1 || echo " [WARN] 迁移再次失败，跳过"
|
|
}
|
|
|
|
USER_COUNT=$(python -c "
|
|
import asyncio
|
|
from app.database import AsyncSessionLocal
|
|
from sqlalchemy import text
|
|
async def check():
|
|
try:
|
|
async with AsyncSessionLocal() as db:
|
|
result = await db.execute(text('SELECT COUNT(*) FROM core_user'))
|
|
print(result.scalar())
|
|
except Exception:
|
|
print(0)
|
|
asyncio.run(check())
|
|
" 2>/dev/null || echo "0")
|
|
|
|
if [ "$USER_COUNT" -eq "0" ] 2>/dev/null; then
|
|
if [ -f "db_init.json" ]; then
|
|
echo " 首次部署，导入初始化数据..."
|
|
python scripts/loaddata.py db_init.json 2>&1
|
|
else
|
|
echo " [WARN] 未找到 db_init.json，跳过数据导入"
|
|
fi
|
|
else
|
|
echo " 数据库已有数据 ($USER_COUNT 个用户)，跳过导入"
|
|
fi
|
|
|
|
echo " 数据库就绪"
|
|
|
|
echo "[4/4] 启动 FastAPI..."
|
|
exec python -m uvicorn main:app --host 0.0.0.0 --port 8000 --workers 4
|