From 254ece76b734553de544cbaa93c5a4ede0e587f5 Mon Sep 17 00:00:00 2001 From: Peter Eisentraut Date: Wed, 10 Aug 2022 23:41:35 +0200 Subject: [PATCH v1 6/7] Enable WRITE_READ_PARSE_PLAN_TREES of rewritten utility statements This was previously disabled because we lacked outfuncs/readfuncs support for most utility statement types. --- src/backend/tcop/postgres.c | 32 +++++++++++--------------------- 1 file changed, 11 insertions(+), 21 deletions(-) diff --git a/src/backend/tcop/postgres.c b/src/backend/tcop/postgres.c index 154e6dbeca..58d2139dc9 100644 --- a/src/backend/tcop/postgres.c +++ b/src/backend/tcop/postgres.c @@ -804,7 +804,7 @@ pg_rewrite_query(Query *query) new_list = copyObject(querytree_list); /* This checks both copyObject() and the equal() routines... */ if (!equal(new_list, querytree_list)) - elog(WARNING, "copyObject() failed to produce equal parse tree"); + elog(WARNING, "copyObject() failed to produce an equal rewritten parse tree"); else querytree_list = new_list; } @@ -816,35 +816,25 @@ pg_rewrite_query(Query *query) List *new_list = NIL; ListCell *lc; - /* - * We currently lack outfuncs/readfuncs support for most utility - * statement types, so only attempt to write/read non-utility queries. - */ foreach(lc, querytree_list) { Query *query = lfirst_node(Query, lc); + char *str = nodeToString(query); + Query *new_query = stringToNodeWithLocations(str); - if (query->commandType != CMD_UTILITY) - { - char *str = nodeToString(query); - Query *new_query = stringToNodeWithLocations(str); - - /* - * queryId is not saved in stored rules, but we must preserve - * it here to avoid breaking pg_stat_statements. - */ - new_query->queryId = query->queryId; + /* + * queryId is not saved in stored rules, but we must preserve + * it here to avoid breaking pg_stat_statements. + */ + new_query->queryId = query->queryId; - new_list = lappend(new_list, new_query); - pfree(str); - } - else - new_list = lappend(new_list, query); + new_list = lappend(new_list, new_query); + pfree(str); } /* This checks both outfuncs/readfuncs and the equal() routines... */ if (!equal(new_list, querytree_list)) - elog(WARNING, "outfuncs/readfuncs failed to produce equal parse tree"); + elog(WARNING, "outfuncs/readfuncs failed to produce an equal rewritten parse tree"); else querytree_list = new_list; } -- 2.37.1