def verify_upgrade(config, mixed_mode):
for port in NODE_PORTS.values():
actual_citus_version = get_actual_citus_version(config.bindir, port)
expected_citus_version = MASTER_VERSION
if expected_citus_version != actual_citus_version and not (mixed_mode and port == WORKER1PORT):
print("port: {} citus version {} expected {}".format(
port, actual_citus_version, expected_citus_version))
sys.exit(1)
else:
print("port:{} citus version {}".format(port, actual_citus_version))
def report_initial_version(config):
for port in NODE_PORTS.values():
actual_citus_version = get_actual_citus_version(config.bindir, port)
print("port:{} citus version {}".format(port, actual_citus_version))
def run_alter_citus(pg_path, mixed_mode):
for port in NODE_PORTS.values():
if mixed_mode and port == WORKER1PORT:
continue
utils.psql(pg_path, port, "ALTER EXTENSION citus UPDATE;")
def create_citus_extension(pg_path):
for port in NODE_PORTS.values():
utils.psql(pg_path, port, "CREATE EXTENSION citus;")
def citus_prepare_pg_upgrade(pg_path):
for port in NODE_PORTS.values():
utils.psql(pg_path, port, "SELECT citus_prepare_pg_upgrade();")
