xuyueandClaude Opus 5.5
127bd6e5ce
Deploy / deploy (push) Canceled after 0s
旧栈 compose 原名 docker-compose.debian.yml，按容器标签里的文件名包含 compose.debian.yml 来认会撞上。「已经搬过」改为只靠「旧库必须是 16」和「postgresql/ 不存在」两道判断。 Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
224 lines
12 KiB
Bash
Executable File
224 lines
12 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
#
|
||
# 一次性：把服务器上的 postgres / redis 从旧栈那份 /root/OJDeploy/docker-compose.yml
|
||
# 搬进 OJ2 的 compose.debian.yml，顺带 PostgreSQL 16 → 18、Redis 7 → 8。
|
||
# 搬完以后服务器上的全部容器都归 OJ2 管，旧 compose 文件里那两个服务就没用了。
|
||
#
|
||
# cd /root/OJDeploy/OJ2
|
||
# docker/move-db-into-oj2.sh # 服务器自己编镜像（慢，但不依赖 CI）
|
||
# docker/move-db-into-oj2.sh --prebuilt # 用 CI rsync 过来的产物
|
||
#
|
||
# ## 跑之前
|
||
#
|
||
# 1. **挑没课的时候。** 两个站点共用这个库，中间两边都用不了。镜像先在旧栈照常服务时
|
||
# 编好，真正停机的只有「备份 → 恢复 → 起栈」那一段。
|
||
# 2. **机房那台先停应用**（库在这边，机房连着它就恢复不了）：
|
||
# docker compose -f docker/compose.school.yml --env-file docker/.env.school stop oj-api oj-worker
|
||
# 没停的话脚本会查出来还有谁连着库，报出 IP 后中止。
|
||
# 3. **docker/.env 的 POSTGRES_PASSWORD 换成强口令**（openssl rand -hex 32）。
|
||
# 原来的口令是默认的 onlinejudge，而这个账号是超级用户、5445 对公网开着 ——
|
||
# 知道默认口令的人就能拿到整个库，还能在容器里执行命令。新库直接用新口令建，
|
||
# 脚本不接受弱口令。机房的 .env.school 要改成同一个值。
|
||
#
|
||
# ## 它做什么
|
||
#
|
||
# 自检 → 编镜像 → 停本机 api/worker → 确认没人连着旧库 → 记行数 → 备份
|
||
# → 停掉并删除旧的 postgres / redis 容器 → 起本栈的 PostgreSQL 18 / Redis 8
|
||
# → 恢复 → 收集统计信息 → 核对行数和网络登录 → docker/deploy.sh --no-build
|
||
#
|
||
# 旧库的数据目录 `$DATA_DIR/postgres`（16）**一个字节都不动**，新库在
|
||
# `$DATA_DIR/postgresql`。Redis 的 `$DATA_DIR/redis` 新旧共用，8 直接读 7 的 dump.rdb。
|
||
#
|
||
# ## 恢复时口令为什么要剥掉
|
||
#
|
||
# pg_dumpall 会带一句 `ALTER ROLE onlinejudge ... PASSWORD 'md5…'`，原样灌进去会把新库
|
||
# 的口令改回旧的 md5 哈希。而 18 的镜像新建的库 pg_hba 只认 scram-sha-256，md5 存的
|
||
# 口令**永远验不过** —— 容器里 docker exec psql 走本地 socket 是 trust，看着一切正常，
|
||
# 只有 api 和机房从网络连进来才炸。所以恢复时把那个 PASSWORD 子句删掉，库里留下的就是
|
||
# 初始化时按 POSTGRES_PASSWORD 存的 SCRAM 哈希。
|
||
#
|
||
# ## 中途失败
|
||
#
|
||
# 停旧容器之前失败：什么都没动，`docker start oj-api oj-worker` 就回去了（容器还是原来的
|
||
# 配置）。别用 `docker compose ... start`：新的 compose 里 api 依赖本栈的 oj-postgres，
|
||
# 这时它还不存在，compose 直接拒绝。
|
||
# 之后失败：旧库目录原样在，回退见 docs/deploy.md「把库搬进 OJ2」。
|
||
|
||
[ -n "${BASH_VERSION:-}" ] || exec bash "$0" "$@"
|
||
|
||
set -euo pipefail
|
||
|
||
cd "$(dirname "${BASH_SOURCE[0]}")/.."
|
||
COMPOSE=(docker compose -f docker/compose.debian.yml --env-file docker/.env)
|
||
|
||
for arg in "$@"; do
|
||
case "$arg" in
|
||
--prebuilt) export ARTIFACTS=prebuilt ;;
|
||
*) echo "未知参数：$arg（可用：--prebuilt）" >&2; exit 2 ;;
|
||
esac
|
||
done
|
||
|
||
say() { printf '\n\033[1;36m==> %s\033[0m\n' "$*"; }
|
||
ok() { printf ' \033[32m✓\033[0m %s\n' "$*"; }
|
||
warn() { printf ' \033[33m!\033[0m %s\n' "$*"; }
|
||
die() { printf '\n\033[1;31m❌ %s\033[0m\n\n' "$*" >&2; exit 1; }
|
||
envget() { grep -E "^$1=" docker/.env | tail -1 | cut -d= -f2- || true; }
|
||
|
||
# ---------------------------------------------------------------- 自检
|
||
say "自检"
|
||
|
||
[ -f docker/.env ] || die "docker/.env 不存在"
|
||
|
||
data_dir=$(envget DATA_DIR)
|
||
[ -n "$data_dir" ] && [ -d "$data_dir" ] || die "DATA_DIR（$data_dir）没设或不是目录。服务器上应该是 /root/OJDeploy/data"
|
||
[ ! -e "$data_dir/postgresql" ] || die "$data_dir/postgresql 已经存在 —— 已经搬过了？
|
||
如果是上次搬到一半失败留下的，确认里面没有要的东西，删掉它再跑。"
|
||
ok "数据目录 $data_dir"
|
||
|
||
pg_pw=$(envget POSTGRES_PASSWORD)
|
||
[ "${#pg_pw}" -ge 16 ] && [ "$pg_pw" != onlinejudge ] || die "docker/.env 的 POSTGRES_PASSWORD 太弱（不到 16 位，或者还是默认的 onlinejudge）。
|
||
这个账号是超级用户、5445 对公网开着。换一个：openssl rand -hex 32
|
||
机房的 docker/.env.school 要改成同一个值。"
|
||
[[ "$pg_pw" =~ ^[A-Za-z0-9._~-]+$ ]] || die "POSTGRES_PASSWORD 里有 URL 不安全的字符（它会拼进 DATABASE_URL），用 openssl rand -hex 32 生成"
|
||
ok "POSTGRES_PASSWORD 是强口令"
|
||
|
||
redis_pw=$(envget REDIS_PASSWORD)
|
||
[ -n "$redis_pw" ] || die "docker/.env 里没有 REDIS_PASSWORD"
|
||
|
||
# 旧容器按发布的端口找，不按名字：旧 compose 起的容器叫什么不归我们管
|
||
old_pg=$(docker ps --filter publish=5445 --format '{{.Names}}' | head -1)
|
||
old_redis=$(docker ps --filter publish=5446 --format '{{.Names}}' | head -1)
|
||
[ -n "$old_pg" ] || die "没找到发布 5445 的容器（旧的 postgres）。它没在跑？"
|
||
[ -n "$old_redis" ] || die "没找到发布 5446 的容器（旧的 redis）。它没在跑？"
|
||
# 「已经搬过了」靠这里的 16 和上面 postgresql/ 不存在来判断，不看容器是哪个 compose 起的：
|
||
# 旧栈那份文件原名 docker-compose.debian.yml，按文件名认会把它当成本栈的（线上撞过）
|
||
pgver=$(docker exec "$old_pg" psql -U onlinejudge -d onlinejudge -tAc 'show server_version_num')
|
||
[ "${pgver:0:2}" = 16 ] || die "$old_pg 是 PostgreSQL $pgver，这个脚本只认 16"
|
||
from() { docker inspect -f '{{.Config.Image}}，{{index .Config.Labels "com.docker.compose.project.config_files"}}' "$1"; }
|
||
ok "旧 postgres：$old_pg（$(from "$old_pg")）"
|
||
ok "旧 redis：$old_redis（$(from "$old_redis")）"
|
||
|
||
redis_src=$(docker inspect -f '{{range .Mounts}}{{if eq .Destination "/data"}}{{.Source}}{{end}}{{end}}' "$old_redis")
|
||
[ -n "$redis_src" ] || die "$old_redis 的 /data 没挂到宿主机目录上，会话和队列搬不过来"
|
||
|
||
db_bytes=$(docker exec "$old_pg" psql -U onlinejudge -d postgres -tAc "select pg_database_size('onlinejudge')")
|
||
avail=$(( $(df -Pk "$data_dir" | awk 'NR==2 {print $4}') * 1024 ))
|
||
[ "$avail" -gt $(( db_bytes * 3 )) ] || die "磁盘余量 $((avail >> 20)) MiB，不到库大小（$((db_bytes >> 20)) MiB）的三倍：新库一份、备份一份，还得留余地"
|
||
ok "库 $((db_bytes >> 20)) MiB，磁盘余量 $((avail >> 20)) MiB"
|
||
|
||
judge_image=$("${COMPOSE[@]}" config | grep -m1 -E '^[[:space:]]*image: oj2-judge' | awk '{print $2}')
|
||
docker image inspect "$judge_image" >/dev/null 2>&1 || die "判题镜像 $judge_image 不在本机，见 docker/judge/README.md"
|
||
|
||
# ---------------------------------------------------------------- 编镜像
|
||
# 旧栈这时还在照常服务，编完才开始停机
|
||
say "编镜像（旧栈照常服务中）"
|
||
"${COMPOSE[@]}" build oj-api oj-worker oj-web
|
||
docker pull -q postgres:18-alpine >/dev/null
|
||
docker pull -q redis:8-alpine >/dev/null
|
||
ok "postgres:18-alpine、redis:8-alpine 已拉取"
|
||
|
||
# ---------------------------------------------------------------- 停机
|
||
say "停本机的 api / worker（从这里开始站点不可用）"
|
||
"${COMPOSE[@]}" stop oj-api oj-worker
|
||
|
||
# 机房的 api / worker 也连着这个库。停没停只能从库这边看
|
||
clients=""
|
||
for _ in $(seq 15); do
|
||
clients=$(docker exec "$old_pg" psql -U onlinejudge -d postgres -tAc \
|
||
"select string_agg(distinct coalesce(host(client_addr), 'local'), ', ')
|
||
from pg_stat_activity
|
||
where backend_type = 'client backend' and pid <> pg_backend_pid()")
|
||
[ -z "$clients" ] && break
|
||
sleep 2
|
||
done
|
||
[ -z "$clients" ] || die "还有客户端连着旧库：$clients
|
||
多半是机房的 api / worker 没停。停掉它们再跑；要先恢复本机服务的话：
|
||
docker start oj-api oj-worker"
|
||
ok "没有客户端连着旧库"
|
||
|
||
counts_sql="select (select count(*) from submission) || ' ' || (select count(*) from \"user\") || ' ' || (select count(*) from problem)"
|
||
before=$(docker exec "$old_pg" psql -U onlinejudge -d onlinejudge -tAc "$counts_sql")
|
||
ok "提交 / 用户 / 题目：$before"
|
||
|
||
# ---------------------------------------------------------------- 备份
|
||
say "备份"
|
||
backup_dir="${BACKUP_DIR:-$(dirname "$PWD")/backups}"
|
||
marker=$(mktemp)
|
||
CONTAINER="$old_pg" BACKUP_DIR="$backup_dir" docker/backup-db.sh
|
||
backup=$(find "$backup_dir" -maxdepth 1 -name 'db_backup_*.sql.gz' -newer "$marker" | sort | tail -1)
|
||
rm -f "$marker"
|
||
[ -n "$backup" ] || die "backup-db.sh 跑完了但找不到新的备份文件（$backup_dir）"
|
||
ok "备份：$backup"
|
||
|
||
# ---------------------------------------------------------------- 换容器
|
||
say "停掉旧的 postgres / redis"
|
||
docker exec -e REDISCLI_AUTH="$redis_pw" "$old_redis" redis-cli save >/dev/null \
|
||
|| warn "redis save 失败（口令不对？），会话可能丢，学生重新登录一次就好"
|
||
docker stop "$old_pg" "$old_redis" >/dev/null
|
||
# 删掉而不只是停：新容器要用同样的名字和端口
|
||
docker rm "$old_pg" "$old_redis" >/dev/null
|
||
ok "已删除容器 $old_pg、$old_redis（数据目录没动）"
|
||
|
||
if [ "$redis_src" != "$data_dir/redis" ]; then
|
||
mkdir -p "$data_dir/redis"
|
||
[ -e "$data_dir/redis/dump.rdb" ] || cp -a "$redis_src/." "$data_dir/redis/"
|
||
ok "redis 数据从 $redis_src 拷到 $data_dir/redis"
|
||
fi
|
||
|
||
say "起 PostgreSQL 18 / Redis 8"
|
||
"${COMPOSE[@]}" up -d oj-postgres oj-redis
|
||
|
||
# 不用 pg_isready 等：初始化期间镜像会起一个只听 socket 的临时实例，pg_isready 走
|
||
# socket 照样说 OK，这时候灌数据会灌进一个马上要被关掉的实例。走网络连就等得到正式的那个，
|
||
# 还顺带验了新口令能不能从网络登录（api 和机房就是这么连的）
|
||
tcp_check() {
|
||
docker exec -e PGPASSWORD="$pg_pw" oj-postgres sh -c \
|
||
'psql -h "$(hostname -i | cut -d" " -f1)" -U onlinejudge -d onlinejudge -tAc "select 1"' >/dev/null 2>&1
|
||
}
|
||
for _ in $(seq 60); do tcp_check && break; sleep 2; done
|
||
tcp_check || die "PostgreSQL 18 两分钟没起来，看 docker logs oj-postgres"
|
||
ok "PostgreSQL 18 起来了，新口令能从网络登录"
|
||
|
||
# ---------------------------------------------------------------- 恢复
|
||
say "恢复"
|
||
log=$(mktemp)
|
||
started=$(date +%s)
|
||
gzip -cd "$backup" \
|
||
| sed -E "/^ALTER ROLE onlinejudge WITH /s/ PASSWORD '[^']*'//" \
|
||
| docker exec -i oj-postgres psql -q -U onlinejudge -d postgres >/dev/null 2>"$log" || true
|
||
# 「先删后建」的 dump 往新实例灌，这两条必然出现（见 backup-db.sh 头部），别的都不该有
|
||
unexpected=$(grep 'ERROR' "$log" | grep -v -e 'current user cannot be dropped' -e 'role "onlinejudge" already exists' || true)
|
||
[ -z "$unexpected" ] || die "恢复时出了预期之外的错（完整日志 $log）：
|
||
$unexpected"
|
||
rm -f "$log"
|
||
ok "恢复完成，用时 $(( $(date +%s) - started ))s"
|
||
|
||
docker exec oj-postgres vacuumdb -U onlinejudge --all --analyze-in-stages -q
|
||
ok "统计信息已收集"
|
||
|
||
after=$(docker exec oj-postgres psql -U onlinejudge -d onlinejudge -tAc "$counts_sql")
|
||
[ "$after" = "$before" ] || die "行数对不上：恢复前 $before，恢复后 $after"
|
||
ok "提交 / 用户 / 题目：$after，和恢复前一致"
|
||
|
||
tcp_check || die "恢复之后新口令从网络登不上了 —— dump 里的口令没剥干净？
|
||
docker exec -it oj-postgres psql -U onlinejudge -d postgres -c '\\password onlinejudge'"
|
||
ok "新口令仍能从网络登录"
|
||
|
||
# ---------------------------------------------------------------- 起栈
|
||
say "起栈（交给 deploy.sh）"
|
||
docker/deploy.sh --no-build
|
||
|
||
cat <<EOF
|
||
|
||
$(printf '\033[1;32m✅ 搬完了\033[0m')
|
||
|
||
还剩三件事：
|
||
|
||
1. 机房：docker/.env.school 的 POSTGRES_PASSWORD 改成和这边一样的新值，然后
|
||
docker compose -f docker/compose.school.yml --env-file docker/.env.school up -d
|
||
2. /root/OJDeploy/docker-compose.yml 里的 oj-postgres / oj-redis 已经没用了，删掉
|
||
（整个文件都没别的服务的话就删文件）。留着的话哪天有人 up 一下，会和本栈抢端口。
|
||
3. 旧库目录 $data_dir/postgres 留着做回退，跑稳一两周再删。
|
||
备份在 $backup
|
||
EOF
|