Enable babelfishpg_tsql.dump_restore GUC while restoring roles (#467)

Set babelfishpg_tsql.dump_restore GUC to true while restoring roles so that we can use this to add special handling during restore. Task: BABEL-5039 Signed-off-by: Rishabh Tanwar [email protected]
babelfish-for-postgresql · Nov 20, 2024 · dedde2a · dedde2a
1 parent 815f17a
commit dedde2a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/bin/pg_dump/dumpall_babel_utils.c b/src/bin/pg_dump/dumpall_babel_utils.c
@@ -144,6 +144,9 @@ dumpBabelRestoreChecks(FILE *OPF, PGconn *conn, int binary_upgrade)
 	res = executeQuery(conn, "SELECT setting::INT from pg_settings WHERE name = 'server_version_num';");
 	source_server_version_num = atoi(PQgetvalue(res, 0, 0));
 
+	/* SET babelfishpg_tsql.dump_restore GUC in the beginning */
+	appendPQExpBufferStr(qry, "SET babelfishpg_tsql.dump_restore = TRUE;\n");
+
 	/*
 	 * Temporarily enable ON_ERROR_STOP so that whole restore script
 	 * execution fails if the following do block raises an error.