#!/usr/bin/env bash
|
# ─────────────────────────────────────────────────────────────────────────────
|
# JNPF PostgreSQL schema 初始化（全新空库专用）
|
# 1) 建四库 jnpf_init / jnpf_flow / jnpf_xxljob / nacos_config
|
# 2) 跑官方初始化脚本（结构，无业务数据）
|
# 3) 跑自建表 DDL self_built_tables_pg.sql
|
# 4) 跑数据接口修正 data_interface_pg_fixes.sql（此刻空表→幂等 no-op，数据迁移后由 migrate-data.sh 再放一次）
|
# 5) 建 Nacos 存储库 schema（nacos_config_pg.sql）
|
#
|
# 前置：cx-postgres 容器已由根 compose 拉起（安全基线随根 compose，本脚本不再部署容器）：
|
# docker volume create postgres_pgdata # 全新机器
|
# docker compose up -d cx-postgres # 在仓库根 / 离线交付包根目录执行
|
#
|
# 安全守卫：目标库已存在且含表时**拒绝执行**（官方 init 含 DROP TABLE IF EXISTS，
|
# 重放会清空既有数据）；确需重灌请显式 FORCE_RESEED=1。
|
#
|
# 用法：
|
# ./install.sh # 建库 + 建表（= schema）
|
# ./install.sh schema # 同上（兼容旧参数写法）
|
#
|
# 参数（环境变量，默认=现场标准值）：
|
# CONTAINER PG 容器名 默认 cx-postgres
|
# PGUSER 超级用户 默认 postgres（exec psql 走容器本地 trust，无需密码）
|
# DB_INIT / DB_FLOW / DB_XXLJOB / DB_NACOS 四库名（测试时可改 *_t8 隔离）
|
# CREATE_EXTENSIONS 是否建常用扩展（1/0），默认 1（非致命，失败仅告警）
|
# FORCE_RESEED 目标库非空时仍强制重灌（1/0），默认 0
|
# ─────────────────────────────────────────────────────────────────────────────
|
set -euo pipefail
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
SQL_DIR="$SCRIPT_DIR/sql"
|
|
CONTAINER="${CONTAINER:-cx-postgres}"
|
PGUSER="${PGUSER:-postgres}"
|
DB_INIT="${DB_INIT:-jnpf_init}"
|
DB_FLOW="${DB_FLOW:-jnpf_flow}"
|
DB_XXLJOB="${DB_XXLJOB:-jnpf_xxljob}"
|
DB_NACOS="${DB_NACOS:-nacos_config}"
|
CREATE_EXTENSIONS="${CREATE_EXTENSIONS:-1}"
|
FORCE_RESEED="${FORCE_RESEED:-0}"
|
STEP="${1:-schema}"
|
|
log() { printf '\033[1;34m[install]\033[0m %s\n' "$*"; }
|
warn() { printf '\033[1;33m[install][warn]\033[0m %s\n' "$*"; }
|
die() { printf '\033[1;31m[install][err]\033[0m %s\n' "$*" >&2; exit 1; }
|
|
# 在容器内跑 psql（本地 unix socket，postgres 超级用户 trust，无需密码/端口）
|
psql_exec() { docker exec -i "$CONTAINER" psql -U "$PGUSER" "$@"; }
|
|
# 等 PG 进程就绪（首次 initdb 数十秒）。此阶段容器健康检查 unhealthy 属预期——
|
# 它探测的业务库正是本脚本要创建的，healthy 只会出现在本脚本成功之后
|
for _i in $(seq 1 60); do
|
if docker exec "$CONTAINER" pg_isready -qU "$PGUSER" 2>/dev/null; then break; fi
|
[ "$_i" = 60 ] && { echo "错误：等待 PG 就绪超时（docker logs $CONTAINER 查看）" >&2; exit 1; }
|
sleep 2
|
done
|
# 把宿主 SQL 文件喂给容器内 psql，遇错即停
|
run_sql_file() {
|
local db="$1" file="$2"
|
[ -f "$file" ] || die "缺 SQL 文件：${file}（git 拉码路线：先跑 ./assemble-sql-onsite.sh；离线包路线：包内 sql/ 应已组装，联系构建机）"
|
log "→ [$db] 执行 $(basename "$file") ($(du -h "$file" | cut -f1))"
|
psql_exec -d "$db" -v ON_ERROR_STOP=1 -q -f - < "$file"
|
}
|
db_exists() { psql_exec -tAc "SELECT 1 FROM pg_database WHERE datname='$1'" | grep -q 1; }
|
create_db() {
|
local db="$1"
|
if db_exists "$db"; then
|
warn "库 $db 已存在，跳过 CREATE DATABASE（幂等）"
|
else
|
log "创建库 $db"
|
psql_exec -d postgres -v ON_ERROR_STOP=1 -c "CREATE DATABASE \"$db\" ENCODING 'UTF8'"
|
fi
|
}
|
# 只数 BASE TABLE（排除 pg_stat_statements 等扩展视图，计数不受 CREATE_EXTENSIONS 影响）
|
table_count() { psql_exec -tAc "SELECT count(*) FROM information_schema.tables WHERE table_schema='public' AND table_type='BASE TABLE'" -d "$1"; }
|
# 全新 schema 基线：官方 293 张表 + 自建 99 张表（含 DMS V2.3 十表及文件对象映射表）。
|
verify_init_table_count() {
|
local init_table_count init_custom_table_count
|
init_table_count="$(table_count "$DB_INIT")"
|
init_custom_table_count=$((init_table_count - 293))
|
if [ "$init_table_count" -ne 392 ] || [ "$init_custom_table_count" -ne 99 ]; then
|
die "表数核对失败：$DB_INIT 共 ${init_table_count} 张 BASE TABLE，自建表 ${init_custom_table_count} 张；期望总数 392 张、自建 99 张"
|
fi
|
}
|
# 非空守卫：init SQL 含 DROP TABLE IF EXISTS，对已有数据的库重放=清库
|
guard_empty() {
|
local db="$1" n
|
db_exists "$db" || return 0
|
n="$(table_count "$db")"
|
[ "$n" -eq 0 ] && return 0
|
if [ "$FORCE_RESEED" = "1" ]; then
|
warn "库 $db 已有 $n 张表，FORCE_RESEED=1 → 继续重灌（既有数据将被 init 脚本覆盖/清空）"
|
else
|
die "库 $db 已有 $n 张表——本脚本只用于全新空库（init 含 DROP TABLE，重放会清数据）。确需重灌：FORCE_RESEED=1 ./install.sh"
|
fi
|
}
|
|
# ── schema：建库 + 官方 init + 自建表 + 数据接口修正 + Nacos 存储库 ──────────
|
do_schema() {
|
docker inspect "$CONTAINER" >/dev/null 2>&1 \
|
|| die "容器 $CONTAINER 不在——先在仓库根/交付包根目录：docker volume create postgres_pgdata && docker compose up -d cx-postgres"
|
|
log "===== 非空守卫 ====="
|
guard_empty "$DB_INIT"; guard_empty "$DB_FLOW"; guard_empty "$DB_XXLJOB"; guard_empty "$DB_NACOS"
|
|
log "===== 建四库 ====="
|
create_db "$DB_INIT"; create_db "$DB_FLOW"; create_db "$DB_XXLJOB"; create_db "$DB_NACOS"
|
|
if [ "$CREATE_EXTENSIONS" = "1" ]; then
|
log "建常用扩展（非致命）"
|
for ext in pg_stat_statements pg_trgm pgcrypto pg_bigm vector; do
|
psql_exec -d "$DB_INIT" -c "CREATE EXTENSION IF NOT EXISTS \"$ext\"" >/dev/null 2>&1 \
|
&& log " ext $ext ✓" || warn " ext $ext 跳过（镜像未含或无需）"
|
done
|
fi
|
|
log "===== 官方初始化脚本 ====="
|
run_sql_file "$DB_INIT" "$SQL_DIR/jnpf_dbnull_init.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/Update_PostgreSQL_1127.sql"
|
run_sql_file "$DB_FLOW" "$SQL_DIR/jnpf_flow_init.sql"
|
run_sql_file "$DB_XXLJOB" "$SQL_DIR/jnpf_xxljob_init.sql"
|
|
log "===== 自建表 DDL ====="
|
run_sql_file "$DB_INIT" "$SQL_DIR/self_built_tables_pg.sql"
|
|
log "===== 自建表增量（迁移快照后的列级对齐，幂等）====="
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_lims_huanjing_20260717.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_lims_sxt_wendingxing_20260717.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_lims_sxt_wendingxing_20260720.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_lims_sxt_flow_biz_status_20260719.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_lims_sxt_quyang_zhongzhi_shenqing_biz_status_20260723.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_lims_sxt_quyang_zhongzhi_shenqing_qingyandan_link_20260723.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_lims_sxt_plan_status_20260720.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_lims_wendingxing_biangeng_20260720.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/create_lims_sxt_quyang_jihua_renwu_20260724.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/alter_lims_sxt_quyang_jihua_renwu_jianyan_xiang_kuaizhao_20260724.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/create_audit_events_ext_20260724.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/create_audit_events_20260724.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_audit_events_title_entry_20260810.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_audit_events_ingest_merge_20260818.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/repair_audit_events_integrity_20260824.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/create_dms_permission_v2_2_20260815.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/upgrade_dms_permission_v2_2_to_v2_3_20260817.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/create_dms_file_objects_20260817.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/upgrade_dms_file_properties_20260820.sql"
|
run_sql_file "$DB_INIT" "$SQL_DIR/update_audit_events_target_timeline_index_20260823.sql"
|
|
log "===== 数据接口修正（空表幂等，数据迁移后 migrate-data.sh 再放一次）====="
|
run_sql_file "$DB_INIT" "$SQL_DIR/data_interface_pg_fixes.sql"
|
|
log "===== Nacos 存储库 schema ====="
|
run_sql_file "$DB_NACOS" "$SQL_DIR/nacos_config_pg.sql"
|
|
log "===== 表数核对 ====="
|
verify_init_table_count
|
log " $DB_INIT BASE TABLE = $(table_count "$DB_INIT") (期望 392：官方293 + 自建99，含 DMS V2.3 十表和文件对象映射表)"
|
log " $DB_FLOW BASE TABLE = $(table_count "$DB_FLOW") (期望 70)"
|
log " $DB_XXLJOB BASE TABLE = $(table_count "$DB_XXLJOB") (期望 11)"
|
log " $DB_NACOS BASE TABLE = $(table_count "$DB_NACOS") (期望 12)"
|
log "schema 就绪。下一步：停机窗口跑 ./migrate-data.sh 迁数据；Nacos 配置内容按 docs/archive/server-pg-cutover-checklist.md SW-7 导入。"
|
}
|
|
case "$STEP" in
|
schema) do_schema ;;
|
container|all) die "步骤 '$STEP' 已废弃：PG 容器统一由根 compose 拉起（docker compose up -d cx-postgres），本脚本只做 schema" ;;
|
*) die "未知步骤 '$STEP'（仅支持 schema）" ;;
|
esac
|
log "install.sh [$STEP] 完成 ✓"
|
