增加一键启动

2026-05-23 08:22:31 +08:00
parent 8631daf6d4
commit aa522936c3
2 changed files with 177 additions and 0 deletions
@@ -0,0 +1,163 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
+COMPOSE_FILE="${ROOT_DIR}/deploy/docker-compose.dev.yml"
+BACKEND_ENV="${ROOT_DIR}/backend/.env"
+BACKEND_ENV_EXAMPLE="${ROOT_DIR}/backend/.env.example"
+
+BACKEND_PID=""
+FRONTEND_PID=""
+CLEANED_UP=0
+
+log() {
+  printf "\033[1;34m[dev]\033[0m %s\n" "$*"
+}
+
+need_cmd() {
+  if ! command -v "$1" >/dev/null 2>&1; then
+    printf "缺少命令：%s\n" "$1" >&2
+    exit 1
+  fi
+}
+
+cleanup() {
+  if [[ "${CLEANED_UP}" == "1" ]]; then
+    return
+  fi
+  CLEANED_UP=1
+
+  log "正在停止前后端开发进程..."
+  if [[ -n "${FRONTEND_PID}" ]] && kill -0 "${FRONTEND_PID}" >/dev/null 2>&1; then
+    kill "${FRONTEND_PID}" >/dev/null 2>&1 || true
+  fi
+  if [[ -n "${BACKEND_PID}" ]] && kill -0 "${BACKEND_PID}" >/dev/null 2>&1; then
+    kill "${BACKEND_PID}" >/dev/null 2>&1 || true
+  fi
+}
+
+wait_container_healthy() {
+  local container_name="$1"
+  local label="$2"
+  local attempt
+
+  log "等待 ${label} 就绪..."
+  for attempt in {1..60}; do
+    local status
+    status="$(docker inspect -f '{{if .State.Health}}{{.State.Health.Status}}{{else}}{{.State.Status}}{{end}}' "${container_name}" 2>/dev/null || true)"
+    if [[ "${status}" == "healthy" || "${status}" == "running" ]]; then
+      log "${label} 已就绪"
+      return 0
+    fi
+    sleep 2
+  done
+
+  printf "%s 启动超时，请检查：docker logs %s\n" "${label}" "${container_name}" >&2
+  exit 1
+}
+
+init_database() {
+  local table_count
+  table_count="$(
+    docker exec hfb-mysql mysql -uhfb -psecret -N -s -e \
+      "SELECT COUNT(*) FROM information_schema.tables WHERE table_schema = 'hfb_sys' AND table_name = 'users';" 2>/dev/null
+  )"
+
+  if [[ "${table_count}" == "0" ]]; then
+    log "首次启动，初始化数据库结构..."
+    docker exec -i hfb-mysql mysql -uhfb -psecret hfb_sys < "${ROOT_DIR}/backend/migrations/000001_init.sql"
+  else
+    log "数据库结构已存在，跳过迁移"
+  fi
+
+  log "刷新开发测试数据..."
+  docker exec -i hfb-mysql mysql -uhfb -psecret hfb_sys < "${ROOT_DIR}/backend/seeds/dev_orders.sql"
+}
+
+load_env_file() {
+  local env_file="$1"
+  local line
+
+  while IFS= read -r line || [[ -n "${line}" ]]; do
+    [[ -z "${line}" || "${line}" =~ ^[[:space:]]*# ]] && continue
+    line="${line#export }"
+    export "${line}"
+  done < "${env_file}"
+}
+
+start_backend() {
+  if [[ ! -f "${BACKEND_ENV}" ]]; then
+    log "创建 backend/.env"
+    cp "${BACKEND_ENV_EXAMPLE}" "${BACKEND_ENV}"
+  fi
+
+  log "启动后端：http://localhost:8080"
+  (
+    cd "${ROOT_DIR}/backend"
+    load_env_file "${BACKEND_ENV}"
+    go run ./cmd/api
+  ) &
+  BACKEND_PID="$!"
+}
+
+start_frontend() {
+  if [[ ! -d "${ROOT_DIR}/frontend/node_modules" ]]; then
+    log "安装前端依赖..."
+    (
+      cd "${ROOT_DIR}/frontend"
+      npm ci
+    )
+  fi
+
+  log "启动前端：http://localhost:5173"
+  (
+    cd "${ROOT_DIR}/frontend"
+    npm run dev
+  ) &
+  FRONTEND_PID="$!"
+}
+
+watch_processes() {
+  while true; do
+    if ! kill -0 "${BACKEND_PID}" >/dev/null 2>&1; then
+      wait "${BACKEND_PID}" || true
+      printf "后端进程已退出\n" >&2
+      exit 1
+    fi
+    if ! kill -0 "${FRONTEND_PID}" >/dev/null 2>&1; then
+      wait "${FRONTEND_PID}" || true
+      printf "前端进程已退出\n" >&2
+      exit 1
+    fi
+    sleep 2
+  done
+}
+
+main() {
+  need_cmd docker
+  need_cmd go
+  need_cmd npm
+
+  trap cleanup EXIT
+  trap 'cleanup; exit 130' INT TERM
+
+  log "启动开发依赖：MySQL、Redis、MinIO"
+  docker compose -f "${COMPOSE_FILE}" up -d
+
+  wait_container_healthy hfb-mysql "MySQL"
+  wait_container_healthy hfb-redis "Redis"
+  wait_container_healthy hfb-minio "MinIO"
+
+  init_database
+  start_backend
+  start_frontend
+
+  log "开发环境已启动"
+  log "前台：http://localhost:5173"
+  log "后台：http://localhost:5173/admin/login（admin / admin123456）"
+  log "按 Ctrl+C 停止前后端；Docker 依赖会保留运行"
+
+  watch_processes
+}
+
+main "$@"