209 lines
7.6 KiB
Bash
Executable File
209 lines
7.6 KiB
Bash
Executable File
#!/bin/bash
|
||
# ===========================================
|
||
# Rust Backend Deployment Script
|
||
# ===========================================
|
||
# Rust Backend Deployment Script
|
||
# ===========================================
|
||
# Usage: ./deploy.sh [development|production] [options]
|
||
#
|
||
# Options:
|
||
# --dry-run 预览模式
|
||
# --yes, -y 跳过确认
|
||
# --skip-tests 跳过部署后测试
|
||
# --rollback 回滚到指定备份
|
||
# --backup-list 列出可用备份
|
||
# --logs [N] 查看后端日志（默认50行）
|
||
# --tail [N] 实时跟踪日志
|
||
# --status 查看服务状态
|
||
# --post-logs 部署后显示日志
|
||
# --init-env 初始化 systemd service 模板
|
||
# --deploy-service 上传 systemd service 文件到服务器
|
||
# --target blue|green 蓝绿部署目标（默认单实例）
|
||
# ===========================================
|
||
|
||
set -euo pipefail
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
source "${SCRIPT_DIR}/lib/common.sh"
|
||
source "${SCRIPT_DIR}/lib/test.sh"
|
||
|
||
# ---------- 默认值 ----------
|
||
PROJECT_NAME="rust-backend"
|
||
REMOTE_USER="root"
|
||
REMOTE_HOST="1panel-server"
|
||
APP_ENV="production"
|
||
|
||
DRY_RUN=false; SKIP_CONFIRM=false; SKIP_TESTS=false
|
||
ROLLBACK=false; LIST_BACKUPS=false; VIEW_LOGS=false; TAIL_LOGS=false
|
||
LOG_LINES=50; VIEW_STATUS=false; POST_DEPLOY_LOGS=false; INIT_ENV=false
|
||
DEPLOY_SERVICE=false; BACKUP_PATH=""
|
||
|
||
# ---------- 选项解析 ----------
|
||
BG_TARGET="" # blue 或 green（仅 production 有效）
|
||
while [[ $# -gt 0 ]]; do
|
||
case "$1" in
|
||
--dry-run) DRY_RUN=true ;;
|
||
--yes|-y) SKIP_CONFIRM=true ;;
|
||
--skip-tests) SKIP_TESTS=true ;;
|
||
--rollback) ROLLBACK=true ;;
|
||
--backup-list) LIST_BACKUPS=true ;;
|
||
--logs) VIEW_LOGS=true ;;
|
||
--tail) TAIL_LOGS=true ;;
|
||
--logs=*) VIEW_LOGS=true; LOG_LINES="${1#*=}" ;;
|
||
--tail=*) TAIL_LOGS=true; LOG_LINES="${1#*=}" ;;
|
||
--status) VIEW_STATUS=true ;;
|
||
--post-logs) POST_DEPLOY_LOGS=true ;;
|
||
--init-env) INIT_ENV=true ;;
|
||
--deploy-service) DEPLOY_SERVICE=true ;;
|
||
--target) BG_TARGET="${2:-}"; shift ;;
|
||
--help|-h)
|
||
echo "用法: $0 [development|production] [options]"
|
||
echo ""
|
||
echo "选项:"
|
||
echo " --dry-run 预览模式"
|
||
echo " --yes, -y 跳过确认"
|
||
echo " --skip-tests 跳过部署后测试"
|
||
echo " --target blue|green 蓝绿部署目标（默认单实例）"
|
||
echo " --rollback 回滚到指定备份"
|
||
echo " --backup-list 列出可用备份"
|
||
echo " --logs [N] 查看后端日志（默认50行）"
|
||
echo " --tail [N] 实时跟踪日志"
|
||
echo " --status 查看服务状态"
|
||
echo " --post-logs 部署后显示日志"
|
||
echo " --init-env 初始化 systemd service 模板"
|
||
echo " --deploy-service 上传 systemd service 文件到服务器"
|
||
exit 0 ;;
|
||
*)
|
||
if [[ "$1" == "development" || "$1" == "production" ]]; then
|
||
APP_ENV="$1"
|
||
else
|
||
echo "未知选项: $1"; exit 1
|
||
fi ;;
|
||
esac; shift
|
||
done
|
||
|
||
# ---------- 蓝绿目标自动检测 + 配置 ----------
|
||
# 优先级: 手动 --target > 自动检测
|
||
detect_blue_green() {
|
||
local proxy_conf; proxy_conf=$(get_proxy_conf)
|
||
local active
|
||
active=$(remote "docker exec ${NGINX_CONTAINER} cat ${proxy_conf} 2>/dev/null" 2>/dev/null | grep -oP '127\.0\.0\.1:\d+' | head -1 || echo "")
|
||
|
||
case "${APP_ENV}:${active}" in
|
||
*:4433) echo "blue" ;;
|
||
*:4434) echo "green" ;;
|
||
*:8083) echo "blue" ;;
|
||
*:8084) echo "green" ;;
|
||
*) echo "" ;;
|
||
esac
|
||
}
|
||
|
||
if [ -z "$BG_TARGET" ] && [ "${APP_ENV}" = "production" ]; then
|
||
# 自动检测：部署到待命环境（仅生产环境支持蓝绿）
|
||
active_target=$(detect_blue_green)
|
||
if [ -n "$active_target" ]; then
|
||
BG_TARGET=$([ "$active_target" = "blue" ] && echo "green" || echo "blue")
|
||
log_info "自动检测: 当前活动 ${active_target}，部署到 ${BG_TARGET}"
|
||
fi
|
||
fi
|
||
|
||
if [ -n "$BG_TARGET" ]; then
|
||
case "${APP_ENV}:${BG_TARGET}" in
|
||
production:blue) BG_PORT="4433"; BG_DIR="rust_backend_blue"; BG_SVC="rust-backend-blue.service" ;;
|
||
production:green) BG_PORT="4434"; BG_DIR="rust_backend_green"; BG_SVC="rust-backend-green.service" ;;
|
||
development:blue) BG_PORT="8083"; BG_DIR="rust_backend_dev_blue"; BG_SVC="rust-backend-dev-blue.service" ;;
|
||
development:green)BG_PORT="8084"; BG_DIR="rust_backend_dev_green";BG_SVC="rust-backend-dev-green.service" ;;
|
||
*) echo "错误: --target 只能是 blue 或 green"; exit 1 ;;
|
||
esac
|
||
fi
|
||
|
||
# ---------- 环境配置 ----------
|
||
case "${APP_ENV}" in
|
||
development)
|
||
if [ -n "$BG_TARGET" ]; then
|
||
REMOTE_DIR="/root/rust/${BG_DIR}"
|
||
SERVICE_NAME="${BG_SVC}"
|
||
BACKEND_PORT="${BG_PORT}"
|
||
else
|
||
REMOTE_DIR="/root/rust/rust_backend_dev"
|
||
SERVICE_NAME="rust-backend-dev.service"
|
||
BACKEND_PORT="8080"
|
||
fi
|
||
TEST_DOMAIN="https://dev.xmclassmate.top"
|
||
DB_CONTAINER="1Panel-postgresql-FtMo"
|
||
DB_NAME="milkydata_dev"
|
||
DB_USER="milkydata" ;;
|
||
production)
|
||
if [ -n "$BG_TARGET" ]; then
|
||
REMOTE_DIR="/root/rust/${BG_DIR}"
|
||
SERVICE_NAME="${BG_SVC}"
|
||
BACKEND_PORT="${BG_PORT}"
|
||
else
|
||
REMOTE_DIR="/root/rust/rust_backend"
|
||
SERVICE_NAME="rust-backend.service"
|
||
BACKEND_PORT="4433"
|
||
fi
|
||
TEST_DOMAIN="https://xmclassmate.top"
|
||
DB_CONTAINER="1Panel-postgresql-FtMo"
|
||
DB_NAME="milkydata"
|
||
DB_USER="milkydata" ;;
|
||
esac
|
||
|
||
# ---------- 快捷命令 ----------
|
||
GIT_VERSION=$(get_git_info)
|
||
ssh_connect
|
||
|
||
if [ "$LIST_BACKUPS" = true ]; then list_backups; exit 0; fi
|
||
if [ "$INIT_ENV" = true ]; then init_env; exit 0; fi
|
||
if [ "$DEPLOY_SERVICE" = true ]; then deploy_service_file; exit 0; fi
|
||
if [ "$VIEW_LOGS" = true ]; then view_logs; exit 0; fi
|
||
if [ "$TAIL_LOGS" = true ]; then tail_logs; exit 0; fi
|
||
if [ "$VIEW_STATUS" = true ]; then status; exit 0; fi
|
||
if [ "$ROLLBACK" = true ]; then rollback; exit 0; fi
|
||
|
||
# ---------- 主部署流程 ----------
|
||
show_deploy_info
|
||
confirm_deploy
|
||
check_not_active
|
||
check_dependencies
|
||
check_service_file
|
||
check_ssl_expiry
|
||
|
||
build_project
|
||
backup_old_version
|
||
cleanup_old_backups 5
|
||
|
||
upload_binary
|
||
upload_dir "迁移文件" migrations
|
||
upload_script "测试脚本" test_deployment.sh test_deployment.sh
|
||
upload_script "清理脚本" scripts/cleanup_refresh_tokens.sh scripts/cleanup_refresh_tokens.sh
|
||
upload_script "备份脚本" scripts/backup-db.sh scripts/backup-db.sh
|
||
|
||
deploy_service_file
|
||
restart_service
|
||
run_migrations
|
||
|
||
if run_tests; then
|
||
log_to_file "DEPLOY success"
|
||
send_webhook "success" "部署完成"
|
||
echo ""; log_info "========== 部署完成！=========="
|
||
echo -e " Git: ${GIT_VERSION}"; echo -e " 环境: ${APP_ENV}"
|
||
if [ -n "$BG_TARGET" ]; then
|
||
echo -e " 目标: ${BG_TARGET} (:${BG_PORT})"
|
||
echo -e ""
|
||
echo -e " ${YELLOW}自动切换代理...${NC}"
|
||
if bash "${SCRIPT_DIR}/scripts/switch-env.sh" --switch "${BG_TARGET}" 2>&1; then
|
||
log_info "代理已切换至 ${BG_TARGET} (:${BG_PORT})"
|
||
else
|
||
log_warn "代理切换失败，请手动执行: ./scripts/switch-env.sh --switch ${BG_TARGET}"
|
||
fi
|
||
fi
|
||
echo ""
|
||
[ "$POST_DEPLOY_LOGS" = true ] && { echo ""; view_logs; }
|
||
else
|
||
log_error "========== 部署失败，已自动回滚 =========="
|
||
exit 1
|
||
fi
|
||
|
||
exit 0
|