优化启动脚本

start_web.sh

@@ -1,5 +1,5 @@
 #!/usr/bin/env bash
-set -e
+set -eo pipefail
 
 ROOT_DIR="$(cd "$(dirname "$0")" && pwd)"
 cd "$ROOT_DIR"
@@ -8,20 +8,38 @@ PYTHON="$ROOT_DIR/.venv/bin/python"
 FRONTEND_DIR="$ROOT_DIR/web/frontend"
 BACKEND_PORT=8000
 FRONTEND_PORT=5173
+HEALTH_TIMEOUT=30  # 后端健康检查最长等待秒数
+
+# ── 工具函数 ──────────────────────────────────────────────
+
+log()  { echo "[$(date '+%H:%M:%S')] $*"; }
+warn() { echo "[$(date '+%H:%M:%S')] ⚠️  $*" >&2; }
+die()  { echo "[$(date '+%H:%M:%S')] ❌ $*" >&2; exit 1; }
+
+check_port() {
+    if lsof -iTCP:"$1" -sTCP:LISTEN -P &>/dev/null; then
+        die "端口 $1 已被占用，请先释放或更改端口号（lsof -i :$1）"
+    fi
+}
+
+# ── 预检查 ────────────────────────────────────────────────
+
+check_port "$BACKEND_PORT"
+check_port "$FRONTEND_PORT"
 
-# 检查 Python 虚拟环境
 if [ ! -f "$PYTHON" ]; then
-    echo "❌ 未找到 .venv，正在创建..."
+    log "未找到 .venv，正在创建..."
     uv venv .venv
     uv pip install -e .
 fi
 
-# 检查前端依赖
 if [ ! -d "$FRONTEND_DIR/node_modules" ]; then
-    echo "❌ 前端依赖未安装，正在安装..."
-    cd "$FRONTEND_DIR" && npm install && cd "$ROOT_DIR"
+    log "前端依赖未安装，正在安装..."
+    (cd "$FRONTEND_DIR" && npm install)
 fi
 
+# ── 启动信息 ──────────────────────────────────────────────
+
 echo ""
 echo "=============================="
 echo "  斗鱼批量登录 - Web 后台"
@@ -33,34 +51,58 @@ echo "  按 Ctrl+C 停止所有服务"
 echo "=============================="
 echo ""
 
-# 清理子进程
+# ── 清理子进程 ────────────────────────────────────────────
+
+PIDS=()
+
 cleanup() {
     echo ""
-    echo "正在停止服务..."
-    kill $BACKEND_PID $FRONTEND_PID 2>/dev/null
-    wait $BACKEND_PID $FRONTEND_PID 2>/dev/null
-    echo "已停止"
+    log "正在停止服务..."
+    for pid in "${PIDS[@]}"; do
+        if kill -0 "$pid" 2>/dev/null; then
+            kill "$pid" 2>/dev/null
+        fi
+    done
+    for pid in "${PIDS[@]}"; do
+        wait "$pid" 2>/dev/null
+    done
+    log "已停止"
 }
 trap cleanup EXIT INT TERM
 
-# 启动后端
-echo "启动后端 (FastAPI :$BACKEND_PORT)..."
+# ── 启动后端 ──────────────────────────────────────────────
+
+log "启动后端 (FastAPI :$BACKEND_PORT)..."
 RELOAD_FLAG=""
 if [ "${UVICORN_RELOAD:-true}" = "true" ]; then
     RELOAD_FLAG="--reload"
 fi
-"$PYTHON" -m uvicorn web.backend.main:app --host 0.0.0.0 --port $BACKEND_PORT $RELOAD_FLAG &
-BACKEND_PID=$!
+"$PYTHON" -m uvicorn web.backend.main:app --host 0.0.0.0 --port "$BACKEND_PORT" $RELOAD_FLAG &
+PIDS+=($!)
 
-# 等后端就绪
-sleep 2
+# 等后端就绪（健康检查循环替代固定 sleep）
+log "等待后端就绪（最多 ${HEALTH_TIMEOUT}s）..."
+elapsed=0
+while [ "$elapsed" -lt "$HEALTH_TIMEOUT" ]; do
+    if curl -sf "http://localhost:$BACKEND_PORT/docs" -o /dev/null 2>/dev/null; then
+        log "后端已就绪 ✓"
+        break
+    fi
+    sleep 1
+    elapsed=$((elapsed + 1))
+done
+if [ "$elapsed" -ge "$HEALTH_TIMEOUT" ]; then
+    warn "后端未在 ${HEALTH_TIMEOUT}s 内就绪，前端可能无法正常连接"
+fi
 
-# 启动前端
-echo "启动前端 (Vite :$FRONTEND_PORT)..."
+# ── 启动前端 ──────────────────────────────────────────────
+
+log "启动前端 (Vite :$FRONTEND_PORT)..."
 cd "$FRONTEND_DIR"
-npm run dev -- --port $FRONTEND_PORT --host 0.0.0.0 &
-FRONTEND_PID=$!
+npm run dev -- --port "$FRONTEND_PORT" --host 0.0.0.0 &
+PIDS+=($!)
 cd "$ROOT_DIR"
 
-# 等待子进程
+# ── 等待子进程 ────────────────────────────────────────────
+
 wait