Update logic

Signed-off-by: Shameem Ahmed <shameemahmed20apr2000@gmail.com>
babelfish-for-postgresql · Jan 10, 2025 · 3d798e9 · 3d798e9
1 parent fed36ee
commit 3d798e9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/contrib/babelfishpg_tsql/src/pltsql_ruleutils.c b/contrib/babelfishpg_tsql/src/pltsql_ruleutils.c
@@ -1434,12 +1434,14 @@ get_rule_expr(Node *node, deparse_context *context,
 				CollateExpr *collate = (CollateExpr *) node;
 				Node	   *arg = (Node *) collate->arg;
 
-				if (!PRETTY_PAREN(context))
+				const char *dump_restore = GetConfigOption("babelfishpg_tsql.dump_restore", true, false);
+
+				if (!PRETTY_PAREN(context) && (!dump_retore || (dump_restore && strcmp(dump_restore, "on") != 0)))
 					appendStringInfoChar(buf, '(');
 				get_rule_expr_paren(arg, context, showimplicit, node);
 				appendStringInfo(buf, " COLLATE %s",
 								 generate_tsql_collation_name(collate->collOid));
-				if (!PRETTY_PAREN(context))
+				if (!PRETTY_PAREN(context) && (!dump_retore || (dump_restore && strcmp(dump_restore, "on") != 0)))
 					appendStringInfoChar(buf, ')');
 			}
 			break;