53 lines
1.9 KiB
Bash
Executable File
53 lines
1.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# 在 SHZ 测试 HighGo 数据库执行指定 SQL 文件。
|
||
# 仅支持测试环境；必须显式设置 CONFIRM_TEST_SQL=yes。
|
||
set -Eeuo pipefail
|
||
|
||
SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd -P)"
|
||
PROJECT_DIR="$(cd -- "$SCRIPT_DIR/../../../.." && pwd -P)"
|
||
|
||
MIGRATION_FILE="${1:-}"
|
||
PASSWORD_FILE="${HIGHGO_TEST_PASSWORD_FILE:-${HIGHGO_PASSWORD_FILE:-$PROJECT_DIR/.codex/skills/query-shz-highgo/.password}}"
|
||
TEST_SSH_HOST="${HIGHGO_TEST_SSH_HOST:-root@47.111.103.66}"
|
||
TEST_SSH_PORT="${HIGHGO_TEST_SSH_PORT:-22}"
|
||
TEST_PSQL_PATH="${HIGHGO_TEST_PSQL_PATH:-/opt/HighGo4.5.7-see/bin/psql}"
|
||
TEST_DB_HOST="${HIGHGO_TEST_DB_HOST:-127.0.0.1}"
|
||
TEST_DB_PORT="${HIGHGO_TEST_DB_PORT:-5866}"
|
||
TEST_DB_NAME="${HIGHGO_TEST_DB_NAME:-highgo}"
|
||
TEST_DB_USER="${HIGHGO_TEST_DB_USER:-sysdba}"
|
||
|
||
if [[ "${CONFIRM_TEST_SQL:-}" != "yes" ]]; then
|
||
echo '为防止误写测试数据库，请使用 CONFIRM_TEST_SQL=yes 执行。' >&2
|
||
exit 2
|
||
fi
|
||
|
||
if [[ -z "$MIGRATION_FILE" ]]; then
|
||
echo "用法：CONFIRM_TEST_SQL=yes $0 <sql-file>" >&2
|
||
exit 2
|
||
fi
|
||
|
||
if [[ ! -r "$MIGRATION_FILE" ]]; then
|
||
echo "SQL 文件不可读：$MIGRATION_FILE" >&2
|
||
exit 2
|
||
fi
|
||
|
||
if [[ ! -r "$PASSWORD_FILE" ]]; then
|
||
echo '测试数据库密码文件不可读。' >&2
|
||
exit 2
|
||
fi
|
||
|
||
command -v ssh >/dev/null 2>&1 || {
|
||
echo '缺少必要命令：ssh' >&2
|
||
exit 2
|
||
}
|
||
|
||
printf '执行测试数据库 SQL：%s\n' "$(basename "$MIGRATION_FILE")"
|
||
{
|
||
cat "$PASSWORD_FILE"
|
||
printf '\n'
|
||
cat "$MIGRATION_FILE"
|
||
} | ssh -o BatchMode=yes -o ConnectTimeout=20 -p "$TEST_SSH_PORT" "$TEST_SSH_HOST" \
|
||
"IFS= read -r PGPASSWORD; export PGPASSWORD; export PGOPTIONS='-c search_path=shz,public -c statement_timeout=30000'; exec '$TEST_PSQL_PATH' -X --no-psqlrc -h '$TEST_DB_HOST' -p '$TEST_DB_PORT' -U '$TEST_DB_USER' -d '$TEST_DB_NAME' -v ON_ERROR_STOP=1 -f -"
|
||
|
||
printf '测试数据库 SQL 执行成功：%s\n' "$(basename "$MIGRATION_FILE")"
|