90 lines
2.9 KiB
Bash
Executable File
90 lines
2.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# 一键部署脚本：打包本地 workspace（排除模型与 .git），上传到远端并重启服务
|
||
|
||
set -euo pipefail
|
||
|
||
SERVER="${SERVER:-120.55.190.8}"
|
||
USER="${USER:-root}"
|
||
# 强烈建议使用 SSH Key；如需密码方式，请通过环境变量传入：
|
||
# export DEPLOY_PASS='your_password'
|
||
PASS="${DEPLOY_PASS:-}"
|
||
REMOTE_DIR="${REMOTE_DIR:-/root/workspace}"
|
||
# 通过环境变量注入 Moonshot Key（不要写死在脚本里）：
|
||
# export MOONSHOT_API_KEY='sk-...'
|
||
MOONSHOT_KEY="${MOONSHOT_API_KEY:-}"
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
ARCHIVE="/tmp/nianjie_deploy_$(date +%Y%m%d%H%M%S).tar.gz"
|
||
|
||
if ! command -v sshpass >/dev/null 2>&1; then
|
||
echo "[ERROR] 需要安装 sshpass 以使用密码方式非交互上传/执行。" >&2
|
||
exit 1
|
||
fi
|
||
if [ -z "${PASS}" ]; then
|
||
echo "[ERROR] 未设置 DEPLOY_PASS（远端 SSH 密码）。" >&2
|
||
exit 1
|
||
fi
|
||
if [ -z "${MOONSHOT_KEY}" ]; then
|
||
echo "[WARN] 未设置 MOONSHOT_API_KEY：云端 AI 模式可能无法调用外部模型。" >&2
|
||
fi
|
||
|
||
echo "[1/5] 打包本地 workspace -> ${ARCHIVE}"
|
||
# macOS: 避免打包扩展属性导致远端 tar 警告
|
||
COPYFILE_DISABLE=1 tar -czf "${ARCHIVE}" \
|
||
--exclude='.git' \
|
||
--exclude='*/.git*' \
|
||
--exclude='minirag/minirag/cache' \
|
||
-C "${SCRIPT_DIR}" .
|
||
|
||
echo "[2/5] 上传压缩包到 ${SERVER}:/root/"
|
||
sshpass -p "${PASS}" scp -o StrictHostKeyChecking=no "${ARCHIVE}" "${USER}@${SERVER}:/root/"
|
||
|
||
echo "[3/5] 远端处理（停服、清理、解压、重启）"
|
||
sshpass -p "${PASS}" ssh -o StrictHostKeyChecking=no "${USER}@${SERVER}" MOONSHOT_KEY="${MOONSHOT_KEY}" bash -s <<'EOF'
|
||
set -euo pipefail
|
||
ARCHIVE="$(ls -t /root/nianjie_deploy_*.tar.gz | head -n1)"
|
||
REMOTE_DIR="/root/workspace"
|
||
MODEL_PATH="${REMOTE_DIR}/minirag/minirag/models"
|
||
BACKUP_DIR="/root/model_backup_$(date +%s)"
|
||
|
||
echo " - 查杀 8081 端口进程"
|
||
if command -v lsof >/dev/null 2>&1; then
|
||
if pids=$(lsof -t -i:8081 || true); then
|
||
[ -n "$pids" ] && kill -9 $pids || true
|
||
fi
|
||
elif command -v fuser >/dev/null 2>&1; then
|
||
fuser -k 8081/tcp || true
|
||
fi
|
||
|
||
echo " - 备份旧模型（如存在）"
|
||
if [ -d "$MODEL_PATH" ]; then
|
||
mkdir -p "$BACKUP_DIR"
|
||
mv "$MODEL_PATH" "$BACKUP_DIR/"
|
||
echo " 备份到 $BACKUP_DIR/models"
|
||
fi
|
||
|
||
echo " - 清空远端 workspace 目录"
|
||
mkdir -p "$REMOTE_DIR"
|
||
rm -rf "$REMOTE_DIR"/*
|
||
|
||
echo " - 解压最新包到 workspace"
|
||
tar -xzf "$ARCHIVE" -C "$REMOTE_DIR"
|
||
|
||
echo " - 恢复模型"
|
||
if [ -d "$BACKUP_DIR/models" ]; then
|
||
mkdir -p "$REMOTE_DIR/minirag/minirag"
|
||
rm -rf "$REMOTE_DIR/minirag/minirag/models"
|
||
mv "$BACKUP_DIR/models" "$REMOTE_DIR/minirag/minirag/" || true
|
||
echo " 模型已恢复"
|
||
fi
|
||
|
||
echo " - 后台重启服务（注入 MOONSHOT_API_KEY）"
|
||
cd "$REMOTE_DIR/dialog"
|
||
export MOONSHOT_API_KEY="${MOONSHOT_KEY:-}"
|
||
nohup python3 app.py > app.log 2>&1 &
|
||
echo " 启动完成，pid: $!"
|
||
EOF
|
||
|
||
echo "[4/5] 部署完成，包保留在 ${ARCHIVE}（本机）与 /root/（远端）"
|
||
echo "[5/5] 结束"
|