94 lines
3.8 KiB
Bash
94 lines
3.8 KiB
Bash
#!/usr/bin/env bash
|
||
# ══════════════════════════════════════════════════
|
||
# 03_start_infra.sh
|
||
# 分步启动基础设施（含健康等待），顺序：
|
||
# PostgreSQL + Redis → etcd + MinIO → Milvus → Neo4j
|
||
# 用法：bash scripts/03_start_infra.sh
|
||
# ══════════════════════════════════════════════════
|
||
set -euo pipefail
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
|
||
cd "$PROJECT_DIR"
|
||
|
||
RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'; BLUE='\033[0;34m'; NC='\033[0m'
|
||
info() { echo -e "${BLUE}[INFO]${NC} $*"; }
|
||
ok() { echo -e "${GREEN}[OK]${NC} $*"; }
|
||
warn() { echo -e "${YELLOW}[WARN]${NC} $*"; }
|
||
error() { echo -e "${RED}[ERROR]${NC} $*"; exit 1; }
|
||
|
||
# 等待服务健康的函数
|
||
wait_healthy() {
|
||
local service=$1
|
||
local max_wait=${2:-120}
|
||
local interval=5
|
||
local elapsed=0
|
||
|
||
info "等待 $service 健康就绪..."
|
||
while [[ $elapsed -lt $max_wait ]]; do
|
||
local status
|
||
status=$(docker compose ps --format json "$service" 2>/dev/null | \
|
||
python3 -c "import sys,json; d=json.load(sys.stdin); print(d.get('Health','unknown'))" 2>/dev/null || echo "unknown")
|
||
|
||
if [[ "$status" == "healthy" ]]; then
|
||
ok "$service 已就绪"
|
||
return 0
|
||
fi
|
||
|
||
echo -n "."
|
||
sleep $interval
|
||
elapsed=$((elapsed + interval))
|
||
done
|
||
echo ""
|
||
error "$service 等待超时（${max_wait}s），请检查：docker compose logs $service"
|
||
}
|
||
|
||
info "══════════════════════════════════════════"
|
||
info " 启动基础设施层"
|
||
info "══════════════════════════════════════════"
|
||
|
||
# ── Step 1：PostgreSQL + Redis ──────────────────
|
||
info "Step 1/4：启动 PostgreSQL 和 Redis..."
|
||
docker compose up -d postgres redis
|
||
|
||
wait_healthy postgres 90
|
||
wait_healthy redis 30
|
||
ok "数据层就绪"
|
||
|
||
# ── Step 2：etcd + MinIO（Milvus 依赖）─────────
|
||
info "Step 2/4：启动 etcd 和 MinIO（Milvus 依赖）..."
|
||
docker compose up -d etcd minio
|
||
|
||
wait_healthy etcd 60
|
||
wait_healthy minio 60
|
||
ok "对象存储层就绪"
|
||
|
||
# ── Step 3：Milvus ──────────────────────────────
|
||
info "Step 3/4：启动 Milvus（向量数据库）..."
|
||
docker compose up -d milvus
|
||
|
||
info "Milvus 初始化需要约 60 秒，请耐心等待..."
|
||
wait_healthy milvus 180
|
||
ok "Milvus 就绪"
|
||
|
||
# ── Step 4：Neo4j ───────────────────────────────
|
||
info "Step 4/4：启动 Neo4j（知识图谱）..."
|
||
docker compose up -d neo4j
|
||
|
||
wait_healthy neo4j 120
|
||
ok "Neo4j 就绪"
|
||
|
||
# ── 汇总 ────────────────────────────────────────
|
||
echo ""
|
||
echo -e "${GREEN}══════════════════════════════════════════${NC}"
|
||
echo -e "${GREEN} 基础设施启动完成！${NC}"
|
||
echo -e "${GREEN}══════════════════════════════════════════${NC}"
|
||
echo ""
|
||
echo " PostgreSQL : localhost:5432"
|
||
echo " Redis : localhost:6379"
|
||
echo " Milvus : localhost:19530 (gRPC), localhost:9091 (HTTP)"
|
||
echo " Neo4j : localhost:7474 (Browser), localhost:7687 (Bolt)"
|
||
echo " MinIO 控制台: localhost:9001 (admin/minioadmin)"
|
||
echo ""
|
||
echo "下一步：bash scripts/04_build_services.sh"
|