selfrelease
64 lines
2.4 KiB
Bash
64 lines
2.4 KiB
Bash
#!/bin/bash
|
|
set -eu
|
|
|
|
# ============================================
|
|
# 远程服务器端部署脚本（由 deploy.sh 调用）
|
|
# 在远程服务器上执行数据库迁移和数据预置
|
|
# ============================================
|
|
|
|
DEMO_DIR="/opt/wxsales/demo"
|
|
|
|
# 查找 psql 路径
|
|
PSQL=""
|
|
for p in /usr/bin/psql /usr/local/bin/psql /usr/pgsql-17/bin/psql /usr/local/pg17/bin/psql; do
|
|
if [ -x "$p" ]; then PSQL="$p"; break; fi
|
|
done
|
|
if [ -z "$PSQL" ]; then
|
|
echo "[ERROR] 未找到 psql"
|
|
exit 1
|
|
fi
|
|
|
|
# 远程环境：PG 15 端口 5432，用 postgres 用户连接
|
|
PSQL_CMD="$PSQL -h 127.0.0.1 -p 5432 -U postgres -d wxchat_sales"
|
|
|
|
# 检查连接方式：优先 su postgres，其次直连
|
|
PG_CONN_OK=0
|
|
su - postgres -c "$PSQL -h 127.0.0.1 -p 5432 -d wxchat_sales -c 'SELECT 1'" >/dev/null 2>&1 && PG_CONN_OK=1
|
|
|
|
if [ "$PG_CONN_OK" -eq 0 ]; then
|
|
echo "[WARN] postgres 用户连接失败，尝试 root 直连..."
|
|
PSQL_CMD="$PSQL -h 127.0.0.1 -p 5432 -d wxchat_sales"
|
|
fi
|
|
|
|
echo "[INFO] 使用 psql: $PSQL"
|
|
echo "[INFO] 执行 schema.sql..."
|
|
|
|
# 执行 schema.sql（幂等，包含迁移语句）
|
|
if [ "$PG_CONN_OK" -eq 1 ]; then
|
|
su - postgres -c "$PSQL -h 127.0.0.1 -p 5432 -d wxchat_sales -f $DEMO_DIR/db/schema.sql" 2>&1 || {
|
|
echo "[WARN] schema.sql 执行有警告，尝试继续..."
|
|
}
|
|
else
|
|
$PSQL_CMD -f "$DEMO_DIR/db/schema.sql" 2>&1 || {
|
|
echo "[WARN] schema.sql 执行有警告，尝试继续..."
|
|
}
|
|
fi
|
|
|
|
# 重置序列
|
|
if [ "$PG_CONN_OK" -eq 1 ]; then
|
|
su - postgres -c "$PSQL -h 127.0.0.1 -p 5432 -d wxchat_sales -c \"SELECT setval('product_id_seq', COALESCE((SELECT max(id) FROM product), 1), true);\"" 2>/dev/null || true
|
|
su - postgres -c "$PSQL -h 127.0.0.1 -p 5432 -d wxchat_sales -c \"SELECT setval('salesperson_id_seq', COALESCE((SELECT max(id) FROM salesperson), 1), true);\"" 2>/dev/null || true
|
|
else
|
|
$PSQL_CMD -c "SELECT setval('product_id_seq', COALESCE((SELECT max(id) FROM product), 1), true);" 2>/dev/null || true
|
|
$PSQL_CMD -c "SELECT setval('salesperson_id_seq', COALESCE((SELECT max(id) FROM salesperson), 1), true);" 2>/dev/null || true
|
|
fi
|
|
|
|
# 验证
|
|
if [ "$PG_CONN_OK" -eq 1 ]; then
|
|
PRODUCT_COUNT=$(su - postgres -c "$PSQL -h 127.0.0.1 -p 5432 -d wxchat_sales -t -c 'SELECT count(*) FROM product;'" 2>/dev/null | xargs || echo "0")
|
|
else
|
|
PRODUCT_COUNT=$($PSQL_CMD -t -c "SELECT count(*) FROM product;" 2>/dev/null | xargs || echo "0")
|
|
fi
|
|
echo "[OK] 产品数量: $PRODUCT_COUNT"
|
|
echo "[OK] 数据库迁移完成"
|