79 lines
4.5 KiB
Bash
79 lines
4.5 KiB
Bash
#!/usr/bin/env bash
|
||
# 切换 A 步（在 47 上跑）：
|
||
# ① 备份 <install-dir>/lib → <platform-dir>/backups/lib-<ts>/
|
||
# ② 覆盖 <install-dir>/lib（T08 集群版代码）
|
||
# ③ 装 **本地 Worker**（<host-a>，19100，无隧道 —— Manager 同机直连）
|
||
# ④ 把既有用户（admin/guest）的归属**预置**为 <host-a>（否则粘性落点无处可粘、新老用户会被按容量随机调度）
|
||
# ⑤ 在 PG 里注册 <host-a> / <host-b> 两台 worker
|
||
set -uo pipefail
|
||
. "$(cd "$(dirname "$0")/.." && pwd)/config/load.sh"
|
||
TS=$(date +%Y%m%d-%H%M%S)
|
||
# 凭据 / 地址一律来自配置（config/platform.env）—— ⛔ 不在脚本里写死
|
||
W47_TOKEN="${DSHS_CLUSTER_AGENT_TOKEN:?config/platform.env 缺 DSHS_CLUSTER_AGENT_TOKEN}"
|
||
W106_TOKEN="${DSH_PEER_AGENT_TOKEN:?config/platform.env 缺 DSH_PEER_AGENT_TOKEN}"
|
||
PGURL="${DSHS_DB_URL:?config/platform.env 缺 DSHS_DB_URL}"
|
||
TARBALL=/tmp/dshs-lib-new.tgz
|
||
|
||
echo "=== ① 备份 $DSH_INSTALL_DIR/lib ==="
|
||
mkdir -p "$DSH_BACKUP_DIR/lib-$TS"
|
||
cp -a "$DSH_INSTALL_DIR/lib" "$DSH_BACKUP_DIR/lib-$TS/lib" && echo " ✓ 备份到 $DSH_BACKUP_DIR/lib-$TS/lib（$(find "$DSH_BACKUP_DIR/lib-$TS" -type f | wc -l) 文件）"
|
||
|
||
echo "=== ② 覆盖 lib ==="
|
||
[ -f "$TARBALL" ] || { echo " ✗ 缺少 $TARBALL"; exit 1; }
|
||
rm -rf "$DSH_INSTALL_DIR"/lib && tar -xzf "$TARBALL" -C "$DSH_INSTALL_DIR"
|
||
echo " ✓ 已覆盖；cluster 特征检查: $(grep -l "DEPLOY_MODE" "$DSH_INSTALL_DIR"/lib/config.js >/dev/null 2>&1 && echo '有 cluster 代码 ✓' || echo '✗ 未见 cluster 代码')"
|
||
echo " lease/agent/tunnel: $(ls "$DSH_INSTALL_DIR"/lib/supervisor/lease.js "$DSH_INSTALL_DIR"/lib/worker/agent.js "$DSH_INSTALL_DIR"/lib/worker/tunnel.js 2>/dev/null | wc -l)/3"
|
||
|
||
echo "=== ③ 本地 Worker 单元（"$DSHS_CLUSTER_HOST_ID"，无隧道） ==="
|
||
cat > /etc/dshs-worker.env <<ENV
|
||
DSHS_DATA_ROOT=$DSHS_DATA_ROOT
|
||
DSHS_ISOLATION_MODE=account
|
||
DSHS_DSH_BIN=/usr/local/bin/dsh
|
||
DSHS_BASE_UID=100000
|
||
DSH_INSTANCE_NODE_OPTIONS=--max-old-space-size=160
|
||
DSH_INSTANCE_UNIVER_SOCKET=auto
|
||
DSHS_CLUSTER_AGENT_TOKEN=$W47_TOKEN
|
||
ENV
|
||
chmod 600 /etc/dshs-worker.env
|
||
cat > /etc/systemd/system/dshs-worker.service <<UNIT
|
||
[Unit]
|
||
Description=DSHS cluster worker agent (this host = 47, local users' instances)
|
||
After=network-online.target
|
||
|
||
[Service]
|
||
Type=simple
|
||
EnvironmentFile=/etc/dshs-worker.env
|
||
ExecStart=/usr/local/bin/node $DSH_INSTALL_DIR/lib/cli.js worker --port 19100 --host 127.0.0.1 --host-id $DSHS_CLUSTER_HOST_ID --instance-host 127.0.0.1 --log-level info
|
||
Restart=on-failure
|
||
RestartSec=3
|
||
KillMode=mixed
|
||
|
||
[Install]
|
||
WantedBy=multi-user.target
|
||
UNIT
|
||
systemctl daemon-reload; systemctl enable dshs-worker >/dev/null 2>&1
|
||
systemctl restart dshs-worker; sleep 5
|
||
echo " dshs-worker=$(systemctl is-active dshs-worker) healthz=$(curl -s -m 6 http://127.0.0.1:19100/healthz | head -c 120)"
|
||
|
||
echo "=== ④ 既有用户归属预置为 $DSHS_CLUSTER_HOST_ID（粘性锚点） ==="
|
||
PGPASSWORD="$DSHS_PG_PASSWORD" /usr/bin/psql -h 127.0.0.1 -p "$DSHS_PG_PORT" -U dshs -d dshs -tAc \
|
||
"insert into dsh_instances (id, user_id, role, status, host_id, epoch, heartbeat_at, lease_until)
|
||
select 'dsh-'||id, id, 'main', 'stopped', '$DSHS_CLUSTER_HOST_ID', 0, 0, 0 from users
|
||
on conflict (id) do update set host_id='$DSHS_CLUSTER_HOST_ID', lease_until=0" 2>&1 | tail -1
|
||
PGPASSWORD="$DSHS_PG_PASSWORD" /usr/bin/psql -h 127.0.0.1 -p "$DSHS_PG_PORT" -U dshs -d dshs -tAc \
|
||
"select u.username||' -> '||coalesce(i.host_id,'NULL') from users u left join dsh_instances i on i.user_id=u.id" 2>&1 | sed 's/^/ /'
|
||
|
||
echo "=== ⑤ 注册两台 worker ==="
|
||
PGPASSWORD="$DSHS_PG_PASSWORD" /usr/bin/psql -h 127.0.0.1 -p "$DSHS_PG_PORT" -U dshs -d dshs -tAc \
|
||
"insert into dsh_hosts (id, endpoint, agent_token, capacity_mb, used_mb, status)
|
||
values ('$DSHS_CLUSTER_HOST_ID','http://127.0.0.1:19100','$W47_TOKEN',1024,0,'up'),
|
||
('$DSH_PEER_HOST_ID','http://127.0.0.1:19000','$W106_TOKEN',2560,0,'up')
|
||
on conflict (id) do update set endpoint=excluded.endpoint, agent_token=excluded.agent_token, capacity_mb=excluded.capacity_mb, status='up'" 2>&1 | tail -1
|
||
PGPASSWORD="$DSHS_PG_PASSWORD" /usr/bin/psql -h 127.0.0.1 -p "$DSHS_PG_PORT" -U dshs -d dshs -tAc \
|
||
"select id||' cap='||capacity_mb||' status='||status||' ep='||endpoint from dsh_hosts order by id" 2>&1 | sed 's/^/ /'
|
||
|
||
echo "=== 回滚剧本（现在就记下） ==="
|
||
echo " rm -f /etc/systemd/system/dshs.service.d/cluster.conf && systemctl daemon-reload && \\"
|
||
echo " systemctl restart dshs # 回 SQLite 单机；lib 回滚 = cp -a $DSH_BACKUP_DIR/lib-$TS/lib $DSH_INSTALL_DIR/lib"
|
||
echo " 备份时间戳: $TS"
|