modules/patroni: pause patroni during upgrade, disable python stdout buffering for migration script

This commit is contained in:
Max Headroom 2023-08-26 22:27:10 +02:00
parent eedae3f8a4
commit 9498e87e9c

View file

@ -361,6 +361,7 @@ in
print(f"setting system identifier to {sysid}") print(f"setting system identifier to {sysid}")
ctl.dcs.initialize(create_new=False, sysid=str(sysid)) ctl.dcs.initialize(create_new=False, sysid=str(sysid))
pg.stop() pg.stop()
os.system("patronictl resume")
''; '';
migrationScript = pkgs.writeShellScript "patroni-migration-replicate-or-self-upgrade" '' migrationScript = pkgs.writeShellScript "patroni-migration-replicate-or-self-upgrade" ''
if [[ "$(consul catalog nodes --service='${cfg.scope}' 2>/dev/null | wc -l)" -gt 0 ]]; then if [[ "$(consul catalog nodes --service='${cfg.scope}' 2>/dev/null | wc -l)" -gt 0 ]]; then
@ -398,6 +399,7 @@ in
fi fi
done done
echo no other nodes are at the target version, performing self-upgrade echo no other nodes are at the target version, performing self-upgrade
patronictl pause
leaderHost="$(patronictl list -f json | jq -r 'map(select(.Role == "Leader" and .State == "running") | .Host) | .[0]')" leaderHost="$(patronictl list -f json | jq -r 'map(select(.Role == "Leader" and .State == "running") | .Host) | .[0]')"
# this is where it gets spicy # this is where it gets spicy
rm -rf '${cfg.postgresqlDataDir}' rm -rf '${cfg.postgresqlDataDir}'
@ -407,6 +409,7 @@ in
export PATRONIMIGRATOR_LEADER_HOST="$leaderHost" export PATRONIMIGRATOR_LEADER_HOST="$leaderHost"
# HACK: find a way to get the port # HACK: find a way to get the port
export PATRONIMIGRATOR_LEADER_PORT="5432" export PATRONIMIGRATOR_LEADER_PORT="5432"
export PYTHONUNBUFFERED=1
exec ${upgradeScript} exec ${upgradeScript}
fi fi
fi fi