219 lines
8.2 KiB
Bash
Executable File
219 lines
8.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# =============================================================================
|
||
# Fischer AgentKit — 生产环境一键启动脚本
|
||
# =============================================================================
|
||
#
|
||
# 启动：Docker Compose（agentkit + Redis + PostgreSQL）
|
||
# 要求：.env 文件存在（含 LLM API Key、JWT Secret）
|
||
#
|
||
# 用法：
|
||
# bash scripts/prod-start.sh # 启动所有服务
|
||
# bash scripts/prod-start.sh --init # 首次启动（初始化 DB）
|
||
# bash scripts/prod-start.sh --build # 构建镜像后启动
|
||
# bash scripts/prod-start.sh --help # 帮助
|
||
# =============================================================================
|
||
|
||
set -euo pipefail
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
PROJECT_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
||
cd "$PROJECT_ROOT"
|
||
|
||
: "${COMPOSE_FILE:=docker-compose.yaml}"
|
||
|
||
# ── 颜色 ────────────────────────────────────────────────────────────────────
|
||
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[1;33m'
|
||
BLUE='\033[0;34m'
|
||
CYAN='\033[0;36m'
|
||
NC='\033[0m'
|
||
|
||
# ── 帮助 ────────────────────────────────────────────────────────────────────
|
||
|
||
show_help() {
|
||
cat <<-'EOF'
|
||
Fischer AgentKit — 生产环境启动
|
||
|
||
用法: bash scripts/prod-start.sh [选项]
|
||
|
||
选项:
|
||
--init 首次启动，初始化数据库表
|
||
--build 先构建/拉取镜像，再启动
|
||
--pull 仅拉取最新镜像（不启动）
|
||
--help 显示此帮助
|
||
|
||
Docker Compose 文件: docker-compose.yaml
|
||
|
||
服务端口:
|
||
8001 — AgentKit API 服务
|
||
6379 — Redis
|
||
5432 — PostgreSQL
|
||
|
||
访问: http://localhost:8001
|
||
EOF
|
||
}
|
||
|
||
# ── 参数解析 ────────────────────────────────────────────────────────────────
|
||
|
||
INIT_DB=0
|
||
BUILD=0
|
||
PULL_ONLY=0
|
||
|
||
while [[ $# -gt 0 ]]; do
|
||
case $1 in
|
||
--init) INIT_DB=1; shift ;;
|
||
--build) BUILD=1; shift ;;
|
||
--pull) PULL_ONLY=1; shift ;;
|
||
--help|-h) show_help; exit 0 ;;
|
||
*) shift ;;
|
||
esac
|
||
done
|
||
|
||
# ── 日志函数 ────────────────────────────────────────────────────────────────
|
||
|
||
info() { echo -e "${BLUE}[INFO]${NC} $*"; }
|
||
ok() { echo -e "${GREEN}[OK]${NC} $*"; }
|
||
warn() { echo -e "${YELLOW}[WARN]${NC} $*"; }
|
||
fail() { echo -e "${RED}[FAIL]${NC} $*"; }
|
||
|
||
# ── 前置检查 ────────────────────────────────────────────────────────────────
|
||
|
||
check_deps() {
|
||
info "检查依赖..."
|
||
for cmd in docker; do
|
||
if ! command -v "$cmd" &>/dev/null; then
|
||
fail "缺少依赖: $cmd"
|
||
exit 1
|
||
fi
|
||
done
|
||
# docker compose v2+ (no separate docker-compose binary needed)
|
||
if ! command -v docker-compose &>/dev/null && docker compose version &>/dev/null; then
|
||
ok "docker compose 可用（v2+）"
|
||
elif command -v docker-compose &>/dev/null; then
|
||
ok "docker-compose 可用"
|
||
else
|
||
fail "缺少 docker 或 docker compose v2"
|
||
exit 1
|
||
fi
|
||
ok "依赖检查通过"
|
||
}
|
||
|
||
check_env() {
|
||
info "检查环境配置..."
|
||
if [[ ! -f .env ]]; then
|
||
fail "未找到 .env 文件"
|
||
echo " 请先创建 .env 文件（参考 .env.example）"
|
||
exit 1
|
||
fi
|
||
# 检查关键变量
|
||
if ! grep -q "DASHSCOPE_API_KEY\|DEEPSEEK_API_KEY" .env 2>/dev/null; then
|
||
warn ".env 中可能缺少 LLM API Key"
|
||
fi
|
||
ok ".env 存在"
|
||
}
|
||
|
||
# ── 数据库初始化 ────────────────────────────────────────────────────────────
|
||
|
||
init_db() {
|
||
info "初始化数据库..."
|
||
docker compose -f "$COMPOSE_FILE" exec -T postgres \
|
||
psql -U agentkit -d agentkit -c "SELECT 1" &>/dev/null || {
|
||
fail "PostgreSQL 容器未就绪，等待后重试..."
|
||
sleep 5
|
||
}
|
||
|
||
# 等待 PostgreSQL 完全就绪
|
||
local attempt=0
|
||
while [[ $attempt -lt 30 ]]; do
|
||
if docker compose -f "$COMPOSE_FILE" exec -T postgres \
|
||
pg_isready -U agentkit &>/dev/null; then
|
||
break
|
||
fi
|
||
sleep 1
|
||
((attempt++))
|
||
done
|
||
|
||
ok "数据库就绪"
|
||
}
|
||
|
||
# ── 主流程 ─────────────────────────────────────────────────────────────────
|
||
|
||
echo ""
|
||
echo -e "${CYAN}═══════════════════════════════════════════════════${NC}"
|
||
echo -e "${CYAN} Fischer AgentKit — 生产环境启动${NC}"
|
||
echo -e "${CYAN}═══════════════════════════════════════════════════${NC}"
|
||
echo ""
|
||
|
||
check_deps
|
||
check_env
|
||
|
||
if [[ $PULL_ONLY -eq 1 ]]; then
|
||
info "拉取最新镜像..."
|
||
docker compose -f "$COMPOSE_FILE" pull
|
||
ok "镜像拉取完成"
|
||
exit 0
|
||
fi
|
||
|
||
echo ""
|
||
echo -e "${GREEN}─────────────────────────────────────────────────${NC}"
|
||
echo -e "${GREEN} 启动 Docker 服务...${NC}"
|
||
echo -e "${GREEN}─────────────────────────────────────────────────${NC}"
|
||
echo ""
|
||
|
||
if [[ $BUILD -eq 1 ]]; then
|
||
info "构建/拉取镜像..."
|
||
docker compose -f "$COMPOSE_FILE" build --pull
|
||
ok "镜像构建完成"
|
||
fi
|
||
|
||
# 创建网络（如不存在）
|
||
docker network ls 2>/dev/null | grep -q "fischer-agentkit_default" || \
|
||
docker network create fischer-agentkit_default &>/dev/null || true
|
||
|
||
info "启动服务（docker compose up -d）..."
|
||
docker compose -f "$COMPOSE_FILE" up -d --remove-orphans
|
||
|
||
# 等待健康检查
|
||
info "等待服务就绪（健康检查最多 120s）..."
|
||
local attempt=0
|
||
while [[ $attempt -lt 60 ]]; do
|
||
if curl -sf http://127.0.0.1:8001/api/v1/health &>/dev/null; then
|
||
break
|
||
fi
|
||
sleep 2
|
||
((attempt++))
|
||
echo -n "."
|
||
done
|
||
echo ""
|
||
|
||
# 显示状态
|
||
echo ""
|
||
echo -e "${GREEN}─────────────────────────────────────────────────${NC}"
|
||
echo -e "${GREEN} Docker 服务状态${NC}"
|
||
echo -e "${GREEN}─────────────────────────────────────────────────${NC}"
|
||
docker compose -f "$COMPOSE_FILE" ps
|
||
|
||
# 状态检查
|
||
if curl -sf http://127.0.0.1:8001/api/v1/health &>/dev/null; then
|
||
echo ""
|
||
ok "AgentKit API 就绪 -> http://127.0.0.1:8001"
|
||
echo ""
|
||
echo -e "${GREEN}═══════════════════════════════════════════════════${NC}"
|
||
echo -e "${GREEN} 生产服务全部启动成功！${NC}"
|
||
echo -e "${GREEN}─────────────────────────────────────────────────${NC}"
|
||
echo -e " AgentKit API: ${GREEN}http://localhost:8001${NC}"
|
||
echo -e " Redis: ${GREEN}localhost:6379${NC}"
|
||
echo -e " PostgreSQL: ${GREEN}localhost:5432${NC}"
|
||
echo -e "${GREEN}─────────────────────────────────────────────────${NC}"
|
||
echo -e " 查看日志: ${CYAN}docker compose -f $COMPOSE_FILE logs -f${NC}"
|
||
echo -e " 停止服务: ${CYAN}bash scripts/prod-stop.sh${NC}"
|
||
echo -e "${GREEN}═══════════════════════════════════════════════════${NC}"
|
||
else
|
||
echo ""
|
||
fail "AgentKit API 启动超时，查看日志诊断："
|
||
echo -e " ${CYAN}docker compose -f $COMPOSE_FILE logs --tail=50${NC}"
|
||
exit 1
|
||
fi
|