413 lines
9.6 KiB
Bash
Executable File
413 lines
9.6 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# 情绪博物馆管理脚本
|
|
# 提供服务管理、监控、备份等功能
|
|
|
|
set -e
|
|
|
|
# 颜色定义
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
YELLOW='\033[1;33m'
|
|
BLUE='\033[0;34m'
|
|
NC='\033[0m'
|
|
|
|
log_info() {
|
|
echo -e "${GREEN}[INFO]${NC} $1"
|
|
}
|
|
|
|
log_warn() {
|
|
echo -e "${YELLOW}[WARN]${NC} $1"
|
|
}
|
|
|
|
log_error() {
|
|
echo -e "${RED}[ERROR]${NC} $1"
|
|
}
|
|
|
|
log_step() {
|
|
echo -e "${BLUE}[STEP]${NC} $1"
|
|
}
|
|
|
|
# 显示帮助信息
|
|
show_help() {
|
|
echo "情绪博物馆管理脚本"
|
|
echo ""
|
|
echo "用法: $0 [命令] [选项]"
|
|
echo ""
|
|
echo "命令:"
|
|
echo " start 启动所有服务"
|
|
echo " stop 停止所有服务"
|
|
echo " restart 重启所有服务"
|
|
echo " status 查看服务状态"
|
|
echo " logs 查看服务日志"
|
|
echo " backup 备份数据"
|
|
echo " restore 恢复数据"
|
|
echo " update 更新服务"
|
|
echo " clean 清理资源"
|
|
echo " monitor 监控服务"
|
|
echo " health 健康检查"
|
|
echo ""
|
|
echo "选项:"
|
|
echo " -f, --follow 跟踪日志输出"
|
|
echo " -s, --service 指定服务名称"
|
|
echo " -h, --help 显示帮助信息"
|
|
echo ""
|
|
echo "示例:"
|
|
echo " $0 start # 启动所有服务"
|
|
echo " $0 logs -f # 跟踪所有服务日志"
|
|
echo " $0 logs -s gateway # 查看网关服务日志"
|
|
echo " $0 restart -s ai-service # 重启AI服务"
|
|
echo ""
|
|
}
|
|
|
|
# 启动服务
|
|
start_services() {
|
|
log_step "启动服务..."
|
|
|
|
if [ -f "docker-compose.prod.yml" ]; then
|
|
docker-compose -f docker-compose.prod.yml up -d
|
|
else
|
|
docker-compose up -d
|
|
fi
|
|
|
|
log_info "服务启动完成"
|
|
sleep 5
|
|
show_status
|
|
}
|
|
|
|
# 停止服务
|
|
stop_services() {
|
|
log_step "停止服务..."
|
|
|
|
if [ -f "docker-compose.prod.yml" ]; then
|
|
docker-compose -f docker-compose.prod.yml down
|
|
else
|
|
docker-compose down
|
|
fi
|
|
|
|
log_info "服务停止完成"
|
|
}
|
|
|
|
# 重启服务
|
|
restart_services() {
|
|
local service_name=${1:-}
|
|
|
|
if [ -n "$service_name" ]; then
|
|
log_step "重启服务: $service_name"
|
|
docker-compose restart "$service_name"
|
|
else
|
|
log_step "重启所有服务..."
|
|
stop_services
|
|
sleep 3
|
|
start_services
|
|
fi
|
|
}
|
|
|
|
# 查看服务状态
|
|
show_status() {
|
|
log_step "服务状态:"
|
|
echo ""
|
|
docker-compose ps
|
|
echo ""
|
|
|
|
# 显示资源使用情况
|
|
log_step "资源使用情况:"
|
|
docker stats --no-stream --format "table {{.Container}}\t{{.CPUPerc}}\t{{.MemUsage}}\t{{.NetIO}}\t{{.BlockIO}}"
|
|
}
|
|
|
|
# 查看日志
|
|
show_logs() {
|
|
local follow_flag=""
|
|
local service_name=""
|
|
|
|
# 解析参数
|
|
while [[ $# -gt 0 ]]; do
|
|
case $1 in
|
|
-f|--follow)
|
|
follow_flag="-f"
|
|
shift
|
|
;;
|
|
-s|--service)
|
|
service_name="$2"
|
|
shift 2
|
|
;;
|
|
*)
|
|
service_name="$1"
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [ -n "$service_name" ]; then
|
|
log_info "查看服务日志: $service_name"
|
|
docker-compose logs $follow_flag "$service_name"
|
|
else
|
|
log_info "查看所有服务日志"
|
|
docker-compose logs $follow_flag
|
|
fi
|
|
}
|
|
|
|
# 备份数据
|
|
backup_data() {
|
|
local backup_dir="backups/$(date +%Y%m%d_%H%M%S)"
|
|
|
|
log_step "开始数据备份..."
|
|
mkdir -p "$backup_dir"
|
|
|
|
# 备份MySQL数据
|
|
log_info "备份MySQL数据..."
|
|
docker-compose exec -T mysql mysqldump -u root -p123456 --all-databases > "$backup_dir/mysql_backup.sql"
|
|
|
|
# 备份Redis数据
|
|
log_info "备份Redis数据..."
|
|
docker-compose exec -T redis redis-cli BGSAVE
|
|
docker cp $(docker-compose ps -q redis):/data/dump.rdb "$backup_dir/redis_backup.rdb"
|
|
|
|
# 备份配置文件
|
|
log_info "备份配置文件..."
|
|
cp -r deploy "$backup_dir/"
|
|
cp docker-compose*.yml "$backup_dir/"
|
|
cp .env "$backup_dir/" 2>/dev/null || true
|
|
|
|
# 压缩备份
|
|
tar -czf "$backup_dir.tar.gz" -C backups "$(basename $backup_dir)"
|
|
rm -rf "$backup_dir"
|
|
|
|
log_info "备份完成: $backup_dir.tar.gz"
|
|
}
|
|
|
|
# 恢复数据
|
|
restore_data() {
|
|
local backup_file="$1"
|
|
|
|
if [ -z "$backup_file" ]; then
|
|
log_error "请指定备份文件"
|
|
echo "用法: $0 restore <backup_file.tar.gz>"
|
|
exit 1
|
|
fi
|
|
|
|
if [ ! -f "$backup_file" ]; then
|
|
log_error "备份文件不存在: $backup_file"
|
|
exit 1
|
|
fi
|
|
|
|
log_step "开始数据恢复..."
|
|
log_warn "此操作将覆盖现有数据，请确认后继续"
|
|
read -p "是否继续? (y/N): " -n 1 -r
|
|
echo
|
|
|
|
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
|
|
log_info "恢复操作已取消"
|
|
exit 0
|
|
fi
|
|
|
|
# 解压备份文件
|
|
local restore_dir="restore_$(date +%Y%m%d_%H%M%S)"
|
|
mkdir -p "$restore_dir"
|
|
tar -xzf "$backup_file" -C "$restore_dir"
|
|
|
|
# 恢复MySQL数据
|
|
log_info "恢复MySQL数据..."
|
|
docker-compose exec -T mysql mysql -u root -p123456 < "$restore_dir"/*/mysql_backup.sql
|
|
|
|
# 恢复Redis数据
|
|
log_info "恢复Redis数据..."
|
|
docker-compose stop redis
|
|
docker cp "$restore_dir"/*/redis_backup.rdb $(docker-compose ps -q redis):/data/dump.rdb
|
|
docker-compose start redis
|
|
|
|
# 清理临时文件
|
|
rm -rf "$restore_dir"
|
|
|
|
log_info "数据恢复完成"
|
|
}
|
|
|
|
# 更新服务
|
|
update_services() {
|
|
log_step "更新服务..."
|
|
|
|
# 拉取最新代码
|
|
if [ -d ".git" ]; then
|
|
log_info "拉取最新代码..."
|
|
git pull
|
|
fi
|
|
|
|
# 重新构建镜像
|
|
log_info "重新构建镜像..."
|
|
docker-compose build --no-cache
|
|
|
|
# 重启服务
|
|
log_info "重启服务..."
|
|
restart_services
|
|
|
|
log_info "服务更新完成"
|
|
}
|
|
|
|
# 清理资源
|
|
clean_resources() {
|
|
log_step "清理Docker资源..."
|
|
|
|
log_warn "此操作将清理未使用的Docker资源"
|
|
read -p "是否继续? (y/N): " -n 1 -r
|
|
echo
|
|
|
|
if [[ $REPLY =~ ^[Yy]$ ]]; then
|
|
# 清理未使用的镜像
|
|
docker image prune -f
|
|
|
|
# 清理未使用的容器
|
|
docker container prune -f
|
|
|
|
# 清理未使用的网络
|
|
docker network prune -f
|
|
|
|
# 清理未使用的卷（谨慎使用）
|
|
# docker volume prune -f
|
|
|
|
log_info "资源清理完成"
|
|
else
|
|
log_info "清理操作已取消"
|
|
fi
|
|
}
|
|
|
|
# 监控服务
|
|
monitor_services() {
|
|
log_step "服务监控面板"
|
|
echo ""
|
|
|
|
while true; do
|
|
clear
|
|
echo "=== 情绪博物馆服务监控 ==="
|
|
echo "时间: $(date)"
|
|
echo ""
|
|
|
|
# 显示服务状态
|
|
echo "📊 服务状态:"
|
|
docker-compose ps
|
|
echo ""
|
|
|
|
# 显示资源使用
|
|
echo "💻 资源使用:"
|
|
docker stats --no-stream --format "table {{.Container}}\t{{.CPUPerc}}\t{{.MemUsage}}\t{{.NetIO}}"
|
|
echo ""
|
|
|
|
# 显示磁盘使用
|
|
echo "💾 磁盘使用:"
|
|
df -h | grep -E "(Filesystem|/dev/)"
|
|
echo ""
|
|
|
|
echo "按 Ctrl+C 退出监控"
|
|
sleep 5
|
|
done
|
|
}
|
|
|
|
# 健康检查
|
|
health_check() {
|
|
log_step "执行健康检查..."
|
|
|
|
local all_healthy=true
|
|
|
|
# 检查MySQL
|
|
if docker-compose exec -T mysql mysqladmin ping -h localhost -u root -p123456 &> /dev/null; then
|
|
log_info "✅ MySQL服务正常"
|
|
else
|
|
log_error "❌ MySQL服务异常"
|
|
all_healthy=false
|
|
fi
|
|
|
|
# 检查Redis
|
|
if docker-compose exec -T redis redis-cli ping | grep -q PONG; then
|
|
log_info "✅ Redis服务正常"
|
|
else
|
|
log_error "❌ Redis服务异常"
|
|
all_healthy=false
|
|
fi
|
|
|
|
# 检查Nacos
|
|
if curl -s http://localhost:8848/nacos/v1/ns/operator/metrics &> /dev/null; then
|
|
log_info "✅ Nacos服务正常"
|
|
else
|
|
log_error "❌ Nacos服务异常"
|
|
all_healthy=false
|
|
fi
|
|
|
|
# 检查网关
|
|
if curl -s http://localhost:9000/actuator/health &> /dev/null; then
|
|
log_info "✅ 网关服务正常"
|
|
else
|
|
log_error "❌ 网关服务异常"
|
|
all_healthy=false
|
|
fi
|
|
|
|
# 检查AI服务
|
|
if curl -s http://localhost:9002/actuator/health &> /dev/null; then
|
|
log_info "✅ AI服务正常"
|
|
else
|
|
log_error "❌ AI服务异常"
|
|
all_healthy=false
|
|
fi
|
|
|
|
# 检查前端
|
|
if curl -s http://localhost:80/health &> /dev/null; then
|
|
log_info "✅ 前端服务正常"
|
|
else
|
|
log_error "❌ 前端服务异常"
|
|
all_healthy=false
|
|
fi
|
|
|
|
if $all_healthy; then
|
|
log_info "🎉 所有服务健康检查通过"
|
|
else
|
|
log_warn "⚠️ 部分服务存在问题，请检查日志"
|
|
fi
|
|
}
|
|
|
|
# 主函数
|
|
main() {
|
|
case "${1:-}" in
|
|
"start")
|
|
start_services
|
|
;;
|
|
"stop")
|
|
stop_services
|
|
;;
|
|
"restart")
|
|
shift
|
|
restart_services "$@"
|
|
;;
|
|
"status")
|
|
show_status
|
|
;;
|
|
"logs")
|
|
shift
|
|
show_logs "$@"
|
|
;;
|
|
"backup")
|
|
backup_data
|
|
;;
|
|
"restore")
|
|
restore_data "$2"
|
|
;;
|
|
"update")
|
|
update_services
|
|
;;
|
|
"clean")
|
|
clean_resources
|
|
;;
|
|
"monitor")
|
|
monitor_services
|
|
;;
|
|
"health")
|
|
health_check
|
|
;;
|
|
"-h"|"--help"|"help")
|
|
show_help
|
|
;;
|
|
*)
|
|
show_help
|
|
;;
|
|
esac
|
|
}
|
|
|
|
main "$@"
|