forked from wangbo/easyai
283 lines
10 KiB
Bash
Executable File
283 lines
10 KiB
Bash
Executable File
#!/bin/bash
|
||
|
||
set -e # 发生错误时终止脚本执行
|
||
|
||
# 仅支持 -h/--help 快速查看
|
||
if [[ "${1:-}" =~ ^(-h|--help)$ ]]; then
|
||
echo "用法: $0"
|
||
echo ""
|
||
echo "默认拉取仓库并更新镜像；也可通过环境变量跳过交互选择。"
|
||
echo ""
|
||
echo "环境变量:"
|
||
echo " UPDATE_MODE=full|image"
|
||
echo " full 更新并拉取仓库（git pull）+ 更新镜像并重启"
|
||
echo " image 仅更新镜像并重启（跳过 git pull）"
|
||
echo " UPDATE_SKIP_REPO_UPDATE=true|false"
|
||
echo " true 等同 UPDATE_MODE=image"
|
||
echo " false 等同 UPDATE_MODE=full"
|
||
echo " UPDATE_HEALTH_TIMEOUT_SECONDS=秒数"
|
||
echo " 等待核心服务恢复可用的最长时间，默认 300 秒"
|
||
echo ""
|
||
echo "仓库更新后会自动执行 scripts/update-env.d 中的幂等环境配置迁移。"
|
||
exit 0
|
||
fi
|
||
|
||
# 进入脚本所在目录（项目根目录）
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]:-$0}")" && pwd)"
|
||
cd "$SCRIPT_DIR"
|
||
|
||
# 命令行内选择更新方式（默认：更新并拉取仓库）
|
||
SKIP_REPO_UPDATE=false
|
||
|
||
normalize_bool() {
|
||
local value
|
||
value="$(printf '%s' "$1" | tr '[:upper:]' '[:lower:]')"
|
||
case "$value" in
|
||
1|true|yes|y|on) echo "true" ;;
|
||
0|false|no|n|off|"") echo "false" ;;
|
||
*)
|
||
echo "❌ 布尔变量值不合法: $1（支持 true/false/1/0/yes/no/y/n/on/off）" >&2
|
||
return 1
|
||
;;
|
||
esac
|
||
}
|
||
|
||
resolve_update_mode() {
|
||
local mode
|
||
if [ -n "${UPDATE_SKIP_REPO_UPDATE:-}" ]; then
|
||
mode="$(normalize_bool "$UPDATE_SKIP_REPO_UPDATE")" || exit 1
|
||
SKIP_REPO_UPDATE="$mode"
|
||
echo "使用环境变量: UPDATE_SKIP_REPO_UPDATE=$SKIP_REPO_UPDATE"
|
||
return
|
||
fi
|
||
|
||
if [ -n "${UPDATE_MODE:-}" ]; then
|
||
mode="$(printf '%s' "$UPDATE_MODE" | tr '[:upper:]' '[:lower:]')"
|
||
case "$mode" in
|
||
full|repo|pull|git|1)
|
||
SKIP_REPO_UPDATE=false
|
||
;;
|
||
image|images|docker|skip-git|skip_repo|skip-repo|2)
|
||
SKIP_REPO_UPDATE=true
|
||
;;
|
||
*)
|
||
echo "❌ UPDATE_MODE 只能是 full 或 image，当前为: $UPDATE_MODE"
|
||
exit 1
|
||
;;
|
||
esac
|
||
echo "使用环境变量: UPDATE_MODE=$UPDATE_MODE"
|
||
return
|
||
fi
|
||
|
||
if [ "${UPDATE_NON_INTERACTIVE:-0}" = "1" ] || [ "${CI:-}" = "true" ] || [ ! -t 0 ]; then
|
||
echo "使用默认更新方式: 更新并拉取仓库（git pull）+ 更新镜像并重启"
|
||
SKIP_REPO_UPDATE=false
|
||
return
|
||
fi
|
||
|
||
echo ""
|
||
echo "请选择更新方式："
|
||
echo " [1] 更新并拉取仓库（git pull）+ 更新镜像并重启（默认）"
|
||
echo " [2] 仅更新镜像并重启（跳过 git pull）"
|
||
read -r -p "请选择 [1/2，回车默认 1]: " choice
|
||
choice="${choice:-1}"
|
||
case "$choice" in
|
||
2) SKIP_REPO_UPDATE=true ;;
|
||
1) ;;
|
||
*) echo "❌ 无效选择，将使用默认：更新并拉取仓库"; SKIP_REPO_UPDATE=false ;;
|
||
esac
|
||
}
|
||
|
||
resolve_update_mode
|
||
|
||
# 拉取整个仓库更新
|
||
if [ "$SKIP_REPO_UPDATE" = false ]; then
|
||
echo ""
|
||
echo "==========================="
|
||
echo "📥 拉取仓库最新代码"
|
||
echo "==========================="
|
||
|
||
if [ ! -d .git ]; then
|
||
echo "❌ 当前目录不是 Git 仓库，无法执行 git pull"
|
||
echo " 请使用 git clone 克隆项目后即可使用 update.sh 更新"
|
||
echo " 克隆命令: git clone https://git.51easyai.com/wangbo/easyai && cd easyai"
|
||
exit 1
|
||
fi
|
||
|
||
before_revision="$(git rev-parse HEAD)"
|
||
echo "📥 正在执行 git pull --ff-only..."
|
||
if git pull --ff-only; then
|
||
echo "✅ 仓库已更新到最新版本"
|
||
else
|
||
echo "❌ git pull 失败，请检查网络或远程仓库配置"
|
||
exit 1
|
||
fi
|
||
|
||
# 确保环境配置文件存在（从 .sample 生成，不覆盖已有文件）
|
||
echo ""
|
||
echo "📝 检查环境配置文件..."
|
||
[ ! -f .env ] && [ -f .env.sample ] && cp .env.sample .env && echo " ✓ .env"
|
||
[ ! -f .env.tools ] && [ -f .env.tools.sample ] && cp .env.tools.sample .env.tools && echo " ✓ .env.tools"
|
||
[ ! -f .env.ASG ] && [ -f .env.ASG.sample ] && cp .env.ASG.sample .env.ASG && echo " ✓ .env.ASG"
|
||
[ ! -f .env.AMS ] && [ -f .env.AMS.sample ] && cp .env.AMS.sample .env.AMS && echo " ✓ .env.AMS"
|
||
|
||
after_revision="$(git rev-parse HEAD)"
|
||
if [ "$before_revision" != "$after_revision" ] && [ "${EASYAI_UPDATE_REEXECUTED:-0}" != "1" ]; then
|
||
echo "🔄 检测到部署仓库已更新，切换到新版更新脚本继续执行..."
|
||
exec env EASYAI_UPDATE_REEXECUTED=1 UPDATE_SKIP_REPO_UPDATE=true "$SCRIPT_DIR/update.sh"
|
||
fi
|
||
|
||
echo ""
|
||
else
|
||
echo "⏭️ 跳过仓库更新，仅更新镜像并重启"
|
||
echo ""
|
||
fi
|
||
|
||
# 升级已有部署时，将空值、历史默认 JWT 密钥和缺失的 WS ticket 密钥
|
||
# 一次性替换为持久化随机值，并确保网关声明 ws_ticket；已有强密钥保持不变。
|
||
if [ ! -f .env ]; then
|
||
echo "❌ 未找到 .env，请先执行 start.sh 完成初始化"
|
||
exit 1
|
||
fi
|
||
|
||
# 每次更新都执行版本化、幂等的环境配置迁移。迁移只补缺失/空值或必要的列表项，
|
||
# 不覆盖用户已经配置的非空值。后续新增必需配置时只需增加编号迁移文件。
|
||
bash ./scripts/apply-update-env-migrations.sh .env "$SCRIPT_DIR"
|
||
|
||
# 老部署缺少绑定配置时，根据现有公开 API 地址迁移：域名/本机保持回环，IP 直连保持可用。
|
||
# shellcheck source=scripts/init-server-http-bind-ip.sh
|
||
. ./scripts/init-server-http-bind-ip.sh
|
||
init_server_http_bind_ip .env
|
||
|
||
# shellcheck source=scripts/init-security-env.sh
|
||
. ./scripts/init-security-env.sh
|
||
init_security_env .env upgrade
|
||
|
||
# 在更新镜像和重建服务前迁移并校验后端公开 API 地址。
|
||
# shellcheck source=scripts/init-public-api-base-url.sh
|
||
. ./scripts/init-public-api-base-url.sh
|
||
init_public_api_base_url .env upgrade "${DEPLOY_PUBLIC_API_BASE_URL:-}"
|
||
|
||
# 浏览器 Origin 为可选项；留空时后端默认允许所有来源，显式配置时才校验。
|
||
# shellcheck source=scripts/init-security-origin.sh
|
||
. ./scripts/init-security-origin.sh
|
||
init_security_origin .env
|
||
|
||
echo "==========================="
|
||
echo "🚀 开始自动安装 Docker 和 Docker Compose"
|
||
echo "==========================="
|
||
|
||
# 检查 Docker 是否已安装
|
||
if command -v docker &> /dev/null; then
|
||
echo "✅ Docker 已安装，跳过安装步骤"
|
||
else
|
||
# 原有的 Docker 安装步骤
|
||
echo "📦 安装依赖..."
|
||
sudo apt update -y
|
||
sudo apt install -y apt-transport-https ca-certificates curl software-properties-common gnupg lsb-release
|
||
|
||
echo "🔑 添加 Docker GPG 密钥..."
|
||
curl -fsSL https://mirrors.nwafu.edu.cn/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
|
||
|
||
echo "🌍 添加 Docker 源..."
|
||
sudo add-apt-repository -y "deb [arch=amd64] https://mirrors.nwafu.edu.cn生成注释ker-ce/linux/ubuntu/ $(lsb_release -cs) stable"
|
||
sudo apt-get update -y
|
||
|
||
echo "🐳 安装 Docker..."
|
||
sudo apt-get install -y docker.io
|
||
|
||
echo "✅ 启动并设置 Docker 开机自启..."
|
||
sudo systemctl enable docker
|
||
sudo systemctl start docker
|
||
fi
|
||
|
||
# 检测并使用正确的 Docker Compose 命令
|
||
# 优先使用新版本的 docker compose（作为 docker 的子命令）
|
||
if docker compose version &> /dev/null; then
|
||
DOCKER_COMPOSE_CMD="docker compose"
|
||
echo "✅ 检测到 Docker Compose (新版本: docker compose)"
|
||
elif command -v docker-compose &> /dev/null; then
|
||
DOCKER_COMPOSE_CMD="docker-compose"
|
||
echo "✅ 检测到 Docker Compose (旧版本: docker-compose)"
|
||
else
|
||
echo "⚙️ 安装 Docker Compose..."
|
||
COMPOSE_BIN_URL="https://static.51easyai.com/docker-compose-linux-x86_64"
|
||
TMP_COMPOSE_BIN="/tmp/docker-compose-linux-x86_64"
|
||
curl -fL "$COMPOSE_BIN_URL" -o "$TMP_COMPOSE_BIN"
|
||
sudo install -m 0755 "$TMP_COMPOSE_BIN" /usr/bin/docker-compose
|
||
rm -f "$TMP_COMPOSE_BIN"
|
||
DOCKER_COMPOSE_CMD="docker-compose"
|
||
echo "✅ Docker Compose 安装完成"
|
||
fi
|
||
|
||
echo "📌 Docker 运行状态:"
|
||
sudo systemctl status docker --no-pager || true
|
||
|
||
echo "📌 Docker Compose 版本:"
|
||
$DOCKER_COMPOSE_CMD version
|
||
|
||
echo "🎉 Docker 和 Docker Compose 已就绪！"
|
||
|
||
wait_for_core_services() {
|
||
local timeout_seconds="${UPDATE_HEALTH_TIMEOUT_SECONDS:-300}"
|
||
local deadline=$((SECONDS + timeout_seconds))
|
||
local services=(easyai-server ws-gateway easyai-web)
|
||
local service container_id state running health all_ready
|
||
|
||
if ! [[ "$timeout_seconds" =~ ^[1-9][0-9]*$ ]]; then
|
||
echo "❌ UPDATE_HEALTH_TIMEOUT_SECONDS 必须是正整数，当前为: $timeout_seconds"
|
||
return 1
|
||
fi
|
||
|
||
echo "⏳ 正在等待核心服务恢复可用（最长 ${timeout_seconds} 秒）..."
|
||
while (( SECONDS < deadline )); do
|
||
all_ready=true
|
||
for service in "${services[@]}"; do
|
||
container_id="$(sudo $DOCKER_COMPOSE_CMD ps -q "$service" 2>/dev/null || true)"
|
||
if [ -z "$container_id" ]; then
|
||
all_ready=false
|
||
continue
|
||
fi
|
||
|
||
state="$(sudo docker inspect --format '{{.State.Running}}|{{if .State.Health}}{{.State.Health.Status}}{{else}}none{{end}}' "$container_id" 2>/dev/null || true)"
|
||
running="${state%%|*}"
|
||
health="${state#*|}"
|
||
if [ "$running" != "true" ] || { [ "$health" != "none" ] && [ "$health" != "healthy" ]; }; then
|
||
all_ready=false
|
||
fi
|
||
done
|
||
|
||
if [ "$all_ready" = true ]; then
|
||
echo "✅ 核心服务已恢复可用"
|
||
return 0
|
||
fi
|
||
sleep 5
|
||
done
|
||
|
||
echo "❌ 更新后核心服务未在 ${timeout_seconds} 秒内恢复可用"
|
||
echo " 当前状态："
|
||
sudo $DOCKER_COMPOSE_CMD ps easyai-server ws-gateway easyai-web || true
|
||
echo " 请运行 '$DOCKER_COMPOSE_CMD logs --tail=100 easyai-server ws-gateway easyai-web' 查看原因"
|
||
return 1
|
||
}
|
||
|
||
# 旧 Redis 若仍把 AOF/RDB 留在容器写层，直接 up 会丢失数据。
|
||
sudo ./scripts/check-redis-persistence.sh
|
||
|
||
echo "🚀 重新启动EasyAI"
|
||
echo "📦 当前磁盘空间："
|
||
df -h "$SCRIPT_DIR" | tail -n 1
|
||
if ! sudo $DOCKER_COMPOSE_CMD pull; then
|
||
echo "❌ 镜像拉取失败，EasyAI 未被标记为更新成功"
|
||
echo " 请检查网络、镜像仓库权限和服务器磁盘空间后重试"
|
||
exit 1
|
||
fi
|
||
if ! sudo $DOCKER_COMPOSE_CMD up -d; then
|
||
echo "❌ 容器启动失败，EasyAI 未被标记为更新成功"
|
||
exit 1
|
||
fi
|
||
if ! wait_for_core_services; then
|
||
exit 1
|
||
fi
|
||
echo "🎉 EasyAI 应用更新成功"
|