external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add debug

m3hm3t/debug-upgrade-test
Mehmet Yilmaz 2025-01-27 11:02:59 +00:00
parent c4734450d1
commit 6d230be55d
1 changed files with 56 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
src/test/regress/citus_tests/upgrade/citus_upgrade_test.py
View File

@ -146,11 +146,66 @@ def restart_database(pg_path, abs_data_path, node_name, node_ports, logfile_pref
subprocess.run(command, check=True)
import subprocess
import utils
def run_alter_citus(pg_path, mixed_mode, config):
for port in config.node_name_to_ports.values():
if mixed_mode and port == config.chosen_random_worker_port:
continue
utils.psql(pg_path, port, "ALTER EXTENSION citus UPDATE;")
alter_extension_sql = """
SELECT '=== BEFORE ALTER ===' AS info;
SELECT pid, query, state, wait_event_type, wait_event
FROM pg_stat_activity
ORDER BY pid;
SELECT l.locktype,
l.relation::regclass AS relname,
l.mode,
l.granted,
a.pid,
a.query AS current_query
FROM pg_locks l
JOIN pg_stat_activity a ON (l.pid = a.pid)
ORDER BY l.relation, l.pid;
ALTER EXTENSION citus UPDATE;
"""
debug_sql = """
SELECT '=== AFTER ALTER ===' AS info;
SELECT pid, query, state, wait_event_type, wait_event
FROM pg_stat_activity
ORDER BY pid;
SELECT l.locktype,
l.relation::regclass AS relname,
l.mode,
l.granted,
a.pid,
a.query AS current_query
FROM pg_locks l
JOIN pg_stat_activity a ON (l.pid = a.pid)
ORDER BY l.relation, l.pid;
"""
try:
utils.psql(pg_path, port, alter_extension_sql)
except subprocess.CalledProcessError as e:
print(f"[ERROR] Failed to run 'ALTER EXTENSION citus UPDATE;' on port {port}")
print(f" psql returned: {e.returncode}, {e.output}")
try:
utils.psql(pg_path, port, debug_sql)
except subprocess.CalledProcessError as inner_e:
print(f"[WARNING] Could not retrieve diagnostic info on port {port}: {inner_e}")
raise
def verify_upgrade(config, mixed_mode, node_ports):