566 lines
17 KiB
Bash
Executable File
566 lines
17 KiB
Bash
Executable File
#!/bin/bash
|
||
#===========================================================================
|
||
# AIOps 智能运维平台 - 统一启动脚本
|
||
# 支持三种运行模式：本地开发 / Docker容器部署 / 独立二进制
|
||
#
|
||
# 用法:
|
||
# ./run.sh dev 本地开发模式（前后端热重载）
|
||
# ./run.sh docker up Docker Compose 容器部署
|
||
# ./run.sh build 编译独立二进制 ./aiops
|
||
# ./run.sh binary 编译并启动独立二进制
|
||
# ./run.sh help 查看帮助
|
||
#===========================================================================
|
||
set -e
|
||
|
||
# ============================================================
|
||
# 配置区
|
||
# ============================================================
|
||
PROJECT_NAME="AIOps 智能运维平台"
|
||
BACKEND_PORT=8080
|
||
FRONTEND_PORT=5173
|
||
BACKEND_DIR="backend"
|
||
FRONTEND_DIR="frontend"
|
||
BACKEND_BIN="aiops-server"
|
||
STANDALONE_BIN="aiops"
|
||
|
||
# 日志配置
|
||
MAX_LOG_FILES=3
|
||
MAX_LOG_SIZE=$((50 * 1024 * 1024)) # 50MB
|
||
|
||
# ============================================================
|
||
# 通用函数
|
||
# ============================================================
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[1;33m'
|
||
BLUE='\033[0;34m'
|
||
CYAN='\033[0;36m'
|
||
NC='\033[0m'
|
||
|
||
info() { echo -e "${BLUE}[INFO]${NC} $1"; }
|
||
success() { echo -e "${GREEN}[OK]${NC} $1"; }
|
||
warning() { echo -e "${YELLOW}[WARN]${NC} $1"; }
|
||
error() { echo -e "${RED}[ERR]${NC} $1"; }
|
||
banner() { echo -e "${CYAN}$1${NC}"; }
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
cd "$SCRIPT_DIR"
|
||
|
||
# PID 文件
|
||
BACKEND_PID_FILE="$SCRIPT_DIR/$BACKEND_DIR/.backend.pid"
|
||
FRONTEND_PID_FILE="$SCRIPT_DIR/$FRONTEND_DIR/.frontend.pid"
|
||
|
||
# 检测操作系统
|
||
detect_os() {
|
||
case "$(uname -s)" in
|
||
Darwin) echo "macos" ;;
|
||
Linux) echo "linux" ;;
|
||
*) echo "unknown" ;;
|
||
esac
|
||
}
|
||
OS=$(detect_os)
|
||
|
||
# 检查端口是否被占用
|
||
check_port() {
|
||
if [ "$OS" = "macos" ]; then
|
||
lsof -ti:$1 2>/dev/null | head -1
|
||
else
|
||
ss -tlnp 2>/dev/null | grep -q ":$1 " && echo "in-use" || echo ""
|
||
fi
|
||
}
|
||
|
||
# 释放端口
|
||
kill_port() {
|
||
local port=$1
|
||
if [ "$OS" = "macos" ]; then
|
||
local pids=$(lsof -ti:$port 2>/dev/null || echo "")
|
||
[ -n "$pids" ] && echo "$pids" | xargs kill -9 2>/dev/null || true
|
||
else
|
||
fuser -k ${port}/tcp 2>/dev/null || true
|
||
fi
|
||
sleep 0.5
|
||
}
|
||
|
||
# 日志轮转
|
||
rotate_log() {
|
||
local log_file=$1
|
||
[ ! -f "$log_file" ] && return 0
|
||
|
||
local file_size
|
||
if [ "$OS" = "macos" ]; then
|
||
file_size=$(stat -f%z "$log_file" 2>/dev/null || echo "0")
|
||
else
|
||
file_size=$(stat -c%s "$log_file" 2>/dev/null || echo "0")
|
||
fi
|
||
|
||
if [ "$file_size" -gt "$MAX_LOG_SIZE" ]; then
|
||
[ -f "${log_file}.${MAX_LOG_FILES}" ] && rm -f "${log_file}.${MAX_LOG_FILES}"
|
||
for i in $(seq $((MAX_LOG_FILES - 1)) -1 1); do
|
||
[ -f "${log_file}.${i}" ] && mv "${log_file}.${i}" "${log_file}.$((i + 1))"
|
||
done
|
||
mv "$log_file" "${log_file}.1"
|
||
info "日志已轮转: $log_file"
|
||
fi
|
||
}
|
||
|
||
# ============================================================
|
||
# 后端服务（本地开发模式）
|
||
# ============================================================
|
||
start_backend() {
|
||
info "启动后端服务..."
|
||
cd "$SCRIPT_DIR/$BACKEND_DIR"
|
||
mkdir -p logs
|
||
rotate_log "logs/server.log"
|
||
|
||
kill_port $BACKEND_PORT
|
||
|
||
export GIN_MODE=debug
|
||
|
||
info "编译 Go 程序..."
|
||
CGO_ENABLED=1 go build -o "$BACKEND_BIN" ./cmd/ 2>&1 || {
|
||
error "编译失败"
|
||
return 1
|
||
}
|
||
|
||
nohup ./"$BACKEND_BIN" >> logs/server.log 2>&1 &
|
||
echo $! > "$BACKEND_PID_FILE"
|
||
|
||
# 等待服务启动
|
||
local retries=0
|
||
while ! curl -s "http://localhost:$BACKEND_PORT/health" > /dev/null 2>&1; do
|
||
retries=$((retries + 1))
|
||
if [ $retries -ge 30 ]; then
|
||
error "后端启动超时，查看日志: $BACKEND_DIR/logs/server.log"
|
||
tail -20 logs/server.log 2>/dev/null
|
||
return 1
|
||
fi
|
||
sleep 1
|
||
done
|
||
|
||
success "后端已启动 (PID: $(cat "$BACKEND_PID_FILE"), 端口: $BACKEND_PORT)"
|
||
return 0
|
||
}
|
||
|
||
stop_backend() {
|
||
if [ -f "$BACKEND_PID_FILE" ]; then
|
||
local pid=$(cat "$BACKEND_PID_FILE" 2>/dev/null || echo "")
|
||
if [ -n "$pid" ] && kill -0 "$pid" 2>/dev/null; then
|
||
kill "$pid" 2>/dev/null || true
|
||
success "后端已停止"
|
||
fi
|
||
rm -f "$BACKEND_PID_FILE"
|
||
fi
|
||
kill_port $BACKEND_PORT
|
||
}
|
||
|
||
# ============================================================
|
||
# 前端服务（本地开发模式）
|
||
# ============================================================
|
||
start_frontend() {
|
||
info "启动前端服务..."
|
||
cd "$SCRIPT_DIR/$FRONTEND_DIR"
|
||
mkdir -p logs
|
||
rotate_log "logs/dev.log"
|
||
|
||
kill_port $FRONTEND_PORT
|
||
|
||
if [ ! -d "node_modules" ]; then
|
||
warning "安装前端依赖..."
|
||
npm install
|
||
fi
|
||
|
||
nohup npm run dev >> logs/dev.log 2>&1 &
|
||
echo $! > "$FRONTEND_PID_FILE"
|
||
|
||
local retries=0
|
||
while ! curl -s "http://localhost:$FRONTEND_PORT" > /dev/null 2>&1; do
|
||
retries=$((retries + 1))
|
||
if [ $retries -ge 30 ]; then
|
||
error "前端启动超时，查看日志: $FRONTEND_DIR/logs/dev.log"
|
||
tail -20 logs/dev.log 2>/dev/null
|
||
return 1
|
||
fi
|
||
sleep 1
|
||
done
|
||
|
||
success "前端已启动 (PID: $(cat "$FRONTEND_PID_FILE"), 端口: $FRONTEND_PORT)"
|
||
return 0
|
||
}
|
||
|
||
stop_frontend() {
|
||
if [ -f "$FRONTEND_PID_FILE" ]; then
|
||
local pid=$(cat "$FRONTEND_PID_FILE" 2>/dev/null || echo "")
|
||
if [ -n "$pid" ] && kill -0 "$pid" 2>/dev/null; then
|
||
kill "$pid" 2>/dev/null || true
|
||
success "前端已停止"
|
||
fi
|
||
rm -f "$FRONTEND_PID_FILE"
|
||
fi
|
||
kill_port $FRONTEND_PORT
|
||
}
|
||
|
||
# ============================================================
|
||
# 本地开发命令
|
||
# ============================================================
|
||
cmd_dev() {
|
||
banner "========== $PROJECT_NAME (本地开发) =========="
|
||
|
||
# 检查依赖
|
||
if ! command -v go &> /dev/null; then
|
||
error "Go 未安装，请先安装 Go 1.21+"
|
||
exit 1
|
||
fi
|
||
if ! command -v npm &> /dev/null; then
|
||
error "npm 未安装，请先安装 Node.js 20+"
|
||
exit 1
|
||
fi
|
||
|
||
if start_backend && start_frontend; then
|
||
echo ""
|
||
banner "========== 所有服务已启动 =========="
|
||
info "前端地址: http://localhost:$FRONTEND_PORT"
|
||
info "后端地址: http://localhost:$BACKEND_PORT"
|
||
info "健康检查: http://localhost:$BACKEND_PORT/health"
|
||
info "数据库: SQLite (backend/sqlite3.db)"
|
||
echo ""
|
||
info "使用 './run.sh stop' 停止服务"
|
||
else
|
||
error "服务启动失败"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
cmd_stop() {
|
||
info "停止本地服务..."
|
||
stop_backend
|
||
stop_frontend
|
||
success "所有本地服务已停止"
|
||
}
|
||
|
||
cmd_restart() {
|
||
cmd_stop
|
||
sleep 1
|
||
cmd_dev
|
||
}
|
||
|
||
cmd_status() {
|
||
info "本地服务状态:"
|
||
if [ -n "$(check_port $BACKEND_PORT)" ]; then
|
||
success "后端: 运行中 (端口 $BACKEND_PORT)"
|
||
else
|
||
warning "后端: 未运行"
|
||
fi
|
||
if [ -n "$(check_port $FRONTEND_PORT)" ]; then
|
||
success "前端: 运行中 (端口 $FRONTEND_PORT)"
|
||
else
|
||
warning "前端: 未运行"
|
||
fi
|
||
if [ -f "$SCRIPT_DIR/$BACKEND_DIR/sqlite3.db" ]; then
|
||
local db_size=$(du -h "$SCRIPT_DIR/$BACKEND_DIR/sqlite3.db" | cut -f1)
|
||
info "数据库: SQLite (sqlite3.db, $db_size)"
|
||
fi
|
||
}
|
||
|
||
cmd_logs() {
|
||
info "实时日志 (Ctrl+C 退出):"
|
||
echo ""
|
||
local pids=()
|
||
if [ -f "$SCRIPT_DIR/$BACKEND_DIR/logs/server.log" ]; then
|
||
echo "=== 后端日志 ==="
|
||
tail -f "$SCRIPT_DIR/$BACKEND_DIR/logs/server.log" &
|
||
pids+=($!)
|
||
fi
|
||
if [ -f "$SCRIPT_DIR/$FRONTEND_DIR/logs/dev.log" ]; then
|
||
echo "=== 前端日志 ==="
|
||
tail -f "$SCRIPT_DIR/$FRONTEND_DIR/logs/dev.log" &
|
||
pids+=($!)
|
||
fi
|
||
trap "kill ${pids[*]} 2>/dev/null; exit 0" INT TERM
|
||
wait
|
||
}
|
||
|
||
# ============================================================
|
||
# Docker Compose 部署
|
||
# ============================================================
|
||
cmd_docker() {
|
||
local action=${1:-help}
|
||
|
||
if ! command -v docker &> /dev/null; then
|
||
error "Docker 未安装，请先安装 Docker"
|
||
exit 1
|
||
fi
|
||
|
||
case "$action" in
|
||
up)
|
||
banner "========== Docker Compose 启动 =========="
|
||
docker compose up -d --build
|
||
echo ""
|
||
info "等待服务就绪..."
|
||
sleep 5
|
||
docker compose ps
|
||
echo ""
|
||
info "前端地址: http://localhost:80"
|
||
info "后端地址: http://localhost:8080"
|
||
info "健康检查: http://localhost:8080/health"
|
||
;;
|
||
down)
|
||
info "停止 Docker Compose..."
|
||
docker compose down
|
||
success "容器已停止"
|
||
;;
|
||
build)
|
||
info "构建 Docker 镜像..."
|
||
docker compose build
|
||
success "镜像构建完成"
|
||
;;
|
||
restart)
|
||
info "重启 Docker Compose..."
|
||
docker compose restart
|
||
success "容器已重启"
|
||
;;
|
||
logs)
|
||
info "Docker 实时日志 (Ctrl+C 退出):"
|
||
docker compose logs -f
|
||
;;
|
||
ps|status)
|
||
docker compose ps
|
||
;;
|
||
help|--help|-h)
|
||
echo "Docker Compose 部署命令:"
|
||
echo " ./run.sh docker up 构建并启动所有容器"
|
||
echo " ./run.sh docker down 停止并移除容器"
|
||
echo " ./run.sh docker build 仅构建镜像"
|
||
echo " ./run.sh docker restart 重启容器"
|
||
echo " ./run.sh docker logs 查看实时日志"
|
||
echo " ./run.sh docker ps 查看容器状态"
|
||
;;
|
||
*)
|
||
error "未知 docker 子命令: $action"
|
||
cmd_docker help
|
||
exit 1
|
||
;;
|
||
esac
|
||
}
|
||
|
||
# ============================================================
|
||
# 编译独立二进制
|
||
# ============================================================
|
||
cmd_build() {
|
||
local target=${1:-binary}
|
||
|
||
case "$target" in
|
||
binary|aiops)
|
||
banner "========== 编译独立二进制 =========="
|
||
cd "$SCRIPT_DIR/$BACKEND_DIR"
|
||
|
||
# 确保 config.yaml 存在（二进制模式默认 SQLite）
|
||
if [ ! -f "config.yaml" ]; then
|
||
info "生成默认 config.yaml (SQLite + 无Redis)..."
|
||
cat > config.yaml << 'YAML'
|
||
mode: "release"
|
||
port: "8080"
|
||
log_level: "info"
|
||
|
||
database:
|
||
driver: "sqlite"
|
||
host: "localhost"
|
||
port: 3306
|
||
user: ""
|
||
password: ""
|
||
dbname: "data/aiops.db"
|
||
charset: "utf8mb4"
|
||
max_conn: 100
|
||
max_idle: 10
|
||
|
||
redis:
|
||
host: "localhost"
|
||
port: 6379
|
||
password: ""
|
||
db: 0
|
||
|
||
jwt:
|
||
secret: "aiops-jwt-secret-change-in-production"
|
||
expire_hour: 24
|
||
|
||
websocket:
|
||
read_buffer_size: 1024
|
||
write_buffer_size: 1024
|
||
YAML
|
||
fi
|
||
|
||
# 静态编译（CGO_ENABLED=1 因为 SQLite 需要 CGO）
|
||
info "编译中..."
|
||
CGO_ENABLED=1 GOPROXY=https://goproxy.cn,direct GONOSUMCHECK='*' GOTOOLCHAIN=go1.22.2 go build -ldflags="-w -s" -o "$STANDALONE_BIN" ./cmd/
|
||
|
||
if [ -f "$STANDALONE_BIN" ]; then
|
||
local bin_size=$(du -h "$STANDALONE_BIN" | cut -f1)
|
||
success "编译完成: $BACKEND_DIR/$STANDALONE_BIN ($bin_size)"
|
||
echo ""
|
||
info "使用方法:"
|
||
echo " ./$BACKEND_DIR/$STANDALONE_BIN # 默认启动 (SQLite, 端口 8080)"
|
||
echo " ./$BACKEND_DIR/$STANDALONE_BIN --port 9090 # 自定义端口"
|
||
echo " ./$BACKEND_DIR/$STANDALONE_BIN --db-driver mysql \\"
|
||
echo " --db-host localhost --db-user root \\"
|
||
echo " --db-password pass --db-name aiops # MySQL 模式"
|
||
echo " ./$BACKEND_DIR/$STANDALONE_BIN --config config.yaml # 使用配置文件"
|
||
else
|
||
error "编译失败"
|
||
exit 1
|
||
fi
|
||
;;
|
||
agent)
|
||
banner "========== 编译 Agent =========="
|
||
cd "$SCRIPT_DIR/agent"
|
||
|
||
info "编译 Linux AMD64..."
|
||
GOOS=linux GOARCH=amd64 CGO_ENABLED=0 go build -ldflags="-w -s" -o aiops-agent-linux-amd64 .
|
||
success "aiops-agent-linux-amd64 ($(du -h aiops-agent-linux-amd64 | cut -f1))"
|
||
|
||
info "编译 Linux ARM64..."
|
||
GOOS=linux GOARCH=arm64 CGO_ENABLED=0 go build -ldflags="-w -s" -o aiops-agent-linux-arm64 .
|
||
success "aiops-agent-linux-arm64 ($(du -h aiops-agent-linux-arm64 | cut -f1))"
|
||
|
||
if [ "$OS" = "macos" ]; then
|
||
info "编译 macOS (当前平台)..."
|
||
CGO_ENABLED=0 go build -ldflags="-w -s" -o aiops-agent .
|
||
success "aiops-agent ($(du -h aiops-agent | cut -f1))"
|
||
fi
|
||
;;
|
||
frontend)
|
||
info "构建前端生产包..."
|
||
cd "$SCRIPT_DIR/$FRONTEND_DIR"
|
||
npm run build
|
||
success "前端构建完成: $FRONTEND_DIR/dist/"
|
||
;;
|
||
help|--help|-h)
|
||
echo "编译命令:"
|
||
echo " ./run.sh build 编译独立二进制 aiops (默认 SQLite)"
|
||
echo " ./run.sh build agent 交叉编译 Agent (amd64+arm64)"
|
||
echo " ./run.sh build frontend 构建前端生产包"
|
||
;;
|
||
*)
|
||
error "未知构建目标: $target"
|
||
cmd_build help
|
||
exit 1
|
||
;;
|
||
esac
|
||
}
|
||
|
||
# ============================================================
|
||
# 独立二进制模式
|
||
# ============================================================
|
||
cmd_binary() {
|
||
cd "$SCRIPT_DIR/$BACKEND_DIR"
|
||
|
||
# 先编译（如果不存在）
|
||
if [ ! -f "$STANDALONE_BIN" ]; then
|
||
info "二进制不存在，先编译..."
|
||
cmd_build binary
|
||
fi
|
||
|
||
banner "========== 启动独立二进制 =========="
|
||
mkdir -p data logs uploads
|
||
|
||
# 传递额外参数
|
||
./"$STANDALONE_BIN" "$@"
|
||
}
|
||
|
||
# ============================================================
|
||
# 帮助
|
||
# ============================================================
|
||
cmd_help() {
|
||
echo ""
|
||
banner " $PROJECT_NAME 启动脚本"
|
||
echo ""
|
||
echo -e " ${GREEN}本地开发${NC}"
|
||
echo " ./run.sh dev 启动前后端开发服务 (SQLite, 端口 8080+5173)"
|
||
echo " ./run.sh stop 停止本地服务"
|
||
echo " ./run.sh restart 重启本地服务"
|
||
echo " ./run.sh status 查看本地服务状态"
|
||
echo " ./run.sh logs 查看实时日志"
|
||
echo ""
|
||
echo -e " ${CYAN}Docker 容器部署${NC}"
|
||
echo " ./run.sh docker up 构建并启动所有容器 (MySQL+Redis+Nginx)"
|
||
echo " ./run.sh docker down 停止并移除容器"
|
||
echo " ./run.sh docker build 仅构建镜像"
|
||
echo " ./run.sh docker restart 重启容器"
|
||
echo " ./run.sh docker logs 查看容器日志"
|
||
echo " ./run.sh docker ps 查看容器状态"
|
||
echo ""
|
||
echo -e " ${YELLOW}编译 & 独立二进制${NC}"
|
||
echo " ./run.sh build 编译独立二进制 ./$BACKEND_DIR/$STANDALONE_BIN"
|
||
echo " ./run.sh binary 编译+启动独立二进制"
|
||
echo " ./run.sh build agent 交叉编译 Agent (amd64+arm64)"
|
||
echo " ./run.sh build frontend 构建前端生产包"
|
||
echo ""
|
||
echo -e " ${BLUE}独立二进制用法${NC}"
|
||
echo " ./$BACKEND_DIR/$STANDALONE_BIN 默认启动 (SQLite, 端口 8080)"
|
||
echo " ./$BACKEND_DIR/$STANDALONE_BIN --port 9090 自定义端口"
|
||
echo " ./$BACKEND_DIR/$STANDALONE_BIN --config config.yaml 使用配置文件"
|
||
echo " ./$BACKEND_DIR/$STANDALONE_BIN --db-driver mysql \\"
|
||
echo " --db-host 127.0.0.1 --db-user root \\"
|
||
echo " --db-password pass --db-name aiops MySQL 模式"
|
||
echo ""
|
||
echo -e " ${GREEN}快捷操作${NC}"
|
||
echo " ./run.sh 等同于 ./run.sh dev"
|
||
echo " ./run.sh help 显示此帮助"
|
||
echo ""
|
||
}
|
||
|
||
# ============================================================
|
||
# 主入口
|
||
# ============================================================
|
||
main() {
|
||
local cmd=${1:-dev}
|
||
|
||
case "$cmd" in
|
||
# 本地开发
|
||
dev|start)
|
||
cmd_dev
|
||
;;
|
||
stop)
|
||
cmd_stop
|
||
;;
|
||
restart)
|
||
cmd_restart
|
||
;;
|
||
status)
|
||
cmd_status
|
||
;;
|
||
logs)
|
||
cmd_logs
|
||
;;
|
||
|
||
# Docker Compose
|
||
docker|compose)
|
||
shift
|
||
cmd_docker "$@"
|
||
;;
|
||
|
||
# 编译
|
||
build)
|
||
shift
|
||
cmd_build "$@"
|
||
;;
|
||
|
||
# 独立二进制
|
||
binary|bin)
|
||
shift
|
||
cmd_binary "$@"
|
||
;;
|
||
|
||
# 帮助
|
||
help|--help|-h)
|
||
cmd_help
|
||
;;
|
||
|
||
*)
|
||
error "未知命令: $cmd"
|
||
cmd_help
|
||
exit 1
|
||
;;
|
||
esac
|
||
}
|
||
|
||
main "$@"
|