set -euo pipefail

cd "$(dirname "$0")"

die() { echo "错误: $*" >&2; exit 1; }

DRY_RUN=0

case "${1:-}" in

"") ;;

--status) DRY_RUN=1 ;;

*) die "未知参数 '$1'(仅支持 --status)" ;;

[[ $# -le 1 ]] || die "多余参数(仅支持一个可选的 --status)"

[[ -f .env ]] || die "找不到 .env(先 cp .env.example .env 并填好)"

env_get() { sed -n "s/^[[:space:]]*$1=//p" .env | tail -1 | sed -e "s/^[\"']//" -e "s/[\"']\$//"; }

IMAGE_TAG="$(env_get IMAGE_TAG)"

IMAGE_REGISTRY="$(env_get IMAGE_REGISTRY)"

[[ -n "$IMAGE_TAG" ]] || die ".env 里 IMAGE_TAG 为空;必须是 git SHA"

[[ "$IMAGE_TAG" =~ ^[0-9a-f]{7,40}$ ]] || die "IMAGE_TAG 必须是 git SHA(当前值: '$IMAGE_TAG');禁止 latest 等可变 tag"

API_IMAGE="${IMAGE_REGISTRY:-local}/xray-api:${IMAGE_TAG}"

docker image inspect "$API_IMAGE" >/dev/null 2>&1 \

|| die "本机没有镜像 $API_IMAGE(save/load 流程:先把 tar 传上来 docker load;registry 流程:先 docker pull $API_IMAGE)"

docker compose ps --status running postgres >/dev/null 2>&1 \

|| die "postgres 容器未运行;先 docker compose up -d --wait postgres"

docker compose exec -T postgres pg_isready -U app -d agent >/dev/null 2>&1 \

|| die "postgres 尚未 ready;用 docker compose up -d --wait postgres 等到 healthy 再重试"

psql_q() {

docker compose exec -T postgres psql -U app -d agent -Atq \

-v ON_ERROR_STOP=1 --set=client_min_messages=warning -c "$1"

}

img_sh() { docker run --rm --entrypoint sh "$API_IMAGE" -c "$1"; }

echo "镜像 : $API_IMAGE"

mapfile -t MIG_DIRS < <(img_sh 'ls -d /workspace/apps/api/*/migrations 2>/dev/null' || true)

[[ ${#MIG_DIRS[@]} -gt 0 ]] || die "镜像内没有找到任何 migrations 目录"

for dir in "${MIG_DIRS[@]}"; do

db="$(basename "$(dirname "$dir")")"

if [[ "$db" != "agent" ]]; then

die "发现迁移目录 $dir,但本脚本目前只配置了 agent 库。

新增数据库时需要在这里补上库名与连接参数(见 deploy/infra-config.json 的 sql_servers)。"

fi

echo "数据库: $db"

# schema_migrations 与 Encore/golang-migrate 同构(单行记当前版本)。

# --status 是纯只读的:表不存在时不建,直接把当前版本视作 0。

if [[ "$(psql_q "SELECT to_regclass('public.schema_migrations') IS NOT NULL;")" == "t" ]]; then

dirty="$(psql_q "SELECT dirty FROM schema_migrations LIMIT 1;")"

[[ "$dirty" == "t" ]] && die "schema_migrations.dirty = true —— 上一次迁移半途失败。

需人工确认库状态后手动清理该标记再重试(本脚本用事务应用,正常不会出现此状态;

出现即说明有人用别的方式改过库)。"

current="$(psql_q "SELECT COALESCE(MAX(version), 0) FROM schema_migrations;")"

else

current=0

(( DRY_RUN )) || psql_q "CREATE TABLE schema_migrations (version bigint NOT NULL PRIMARY KEY, dirty boolean NOT NULL);" >/dev/null

fi

current="${current:-0}"

echo "当前版本: $current"

mapfile -t FILES < <(img_sh "ls -1 $dir/*.up.sql 2>/dev/null" || true)

[[ ${#FILES[@]} -gt 0 ]] || die "$dir 下没有 *.up.sql"

# 版本号解析 + 数值排序 + 唯一性校验(codex 复审 2026-08-31 第 2 轮 P2):

# - 字典序 sort 在版本宽度变化时会错序(999 排在 1000 后 → v999 被静默跳过);

# - 两个分支各自新增同号迁移(002_a / 002_b),第一份应用后第二份会被

# 「version > current」判断静默跳过,库里永久缺一份变更。

# 两种情况都必须在执行前拒绝,而不是靠约定。

ORDERED=()

for f in "${FILES[@]}"; do

base="$(basename "$f")"

# 文件名前缀数字即版本号(001_init.up.sql -> 1)

ver="$(printf '%s' "$base" | sed -n 's/^0*\([0-9][0-9]*\)_.*/\1/p')"

[[ -n "$ver" ]] || die "迁移文件名不符合 <数字>_<名字>.up.sql 约定: $base"

ORDERED+=("$ver $f")

done

mapfile -t ORDERED < <(printf '%s\n' "${ORDERED[@]}" | sort -n -k1,1)

dup="$(printf '%s\n' "${ORDERED[@]}" | awk '{print $1}' | uniq -d | head -1)"

[[ -z "$dup" ]] || die "发现重复的迁移版本号 v$dup(多个分支各自新增了同号迁移?)。重命名消除冲突后再部署。"

pending=0

for entry in "${ORDERED[@]}"; do

ver="${entry%% *}"

f="${entry#* }"

base="$(basename "$f")"

(( ver > current )) || continue

pending=$((pending + 1))

if (( DRY_RUN )); then

echo " 待执行: v$ver $base"

continue

fi

sql="$(img_sh "cat '$f'")"

# CREATE INDEX CONCURRENTLY 等语句不能在事务里跑;宁可报错也不要绕过事务保护

if printf '%s' "$sql" | grep -qiE '\bCONCURRENTLY\b'; then

die "$base 含 CONCURRENTLY,无法在事务内执行。

该迁移需人工按停机窗口单独处理,处理完手动把 schema_migrations 推到 v$ver。"

fi

echo " 应用 v$ver $base"

# 单事务:SQL 与版本推进同生共死。失败则整体回滚,版本号不动,可直接重跑。

# 并发保护(codex 复审 2026-08-31 P2):advisory 事务锁串行化所有执行者,

# 锁下复核版本——若另一执行者已推进版本,RAISE 中止本事务而不是重复应用。

# 锁 key 为任取的项目常数,随事务结束自动释放,不需要手动解锁。

{

echo "BEGIN;"

# PERFORM 而不是 SELECT:后者会让 psql 把锁函数的返回值当结果集打印出来

# (每个迁移一个空的 "pg_advisory_xact_lock" 表格),把真正的进度行淹掉。

# R9 在 130 上首次执行 6 个迁移时看到这堆噪音,纯观感问题,行为不变。

echo 'DO $lock$ BEGIN PERFORM pg_advisory_xact_lock(823567001); END $lock$;'

echo 'DO $mig$ BEGIN'

echo " IF (SELECT COALESCE(MAX(version), 0) FROM schema_migrations) <> $current THEN"

echo " RAISE EXCEPTION '并发迁移: schema_migrations 当前版本与预期 $current 不符,另一执行者可能正在迁移';"

echo " END IF;"

echo 'END $mig$;'

printf '%s\n' "$sql"

echo "DELETE FROM schema_migrations;"

echo "INSERT INTO schema_migrations (version, dirty) VALUES ($ver, false);"

echo "COMMIT;"

} | docker compose exec -T postgres psql -U app -d agent -v ON_ERROR_STOP=1 -q \

|| die "v$ver 应用失败,已回滚(版本号仍为 $current)。修好后重跑本脚本。"

current="$ver"

done

if (( pending == 0 )); then

echo " 无待执行迁移(已是最新)"

elif (( DRY_RUN )); then

echo " 共 $pending 个待执行(--status 模式,未改库)"

else

echo " 完成,当前版本: $(psql_q "SELECT MAX(version) FROM schema_migrations;")"

fi